维 Tensor(矩阵)

在 pytorch 中，2 维矩阵主要用于表示图片，在计算机中，我们是用 2 维矩阵表示图片每一个像素位置，然后在通过一个通道维度表示像素的颜色和透明度的信息。
在 mnist 数据集为例来说明 2 维矩阵在神经网络中的用途，输入图片为 

，然后将 2 维图图像展平为 1 维 784 维，然后如果一次输入多长图片，就是

 表示每一个批次为 32

a = torch.rand(2,3)