小编Spa*_*Man的帖子

为什么 MNIST 图像是 1x28x28 张量?

我将 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 有什么意义呢?

python machine-learning mnist pytorch

2
推荐指数
1
解决办法
3308
查看次数

标签 统计

machine-learning ×1

mnist ×1

python ×1

pytorch ×1