我将 28x28 像素图像的 MNIST 图像转换为张量
dataset = MNIST(root='data/', train=True, transform=transforms.ToTensor())
Run Code Online (Sandbox Code Playgroud)
当我跑步时
img_tensor, label = dataset[0]
print(img_tensor.shape, label)
Run Code Online (Sandbox Code Playgroud)
据说形状是torch.Size([1, 28, 28])。为什么是 1x28x28?第一维是什么意思?1x28x28 相对于 28x28 有什么意义呢?