深度学习：PyTorch的张量数据类型

import numpy as np
import torch

#示例一
a=torch.rand(2,3,28,28)#2表示两张图片，3表示三个通道数，RGB图片，28*28是图片的大小
print(a.numel())#2*3*28*28
print(a.dim())#dimension=4

#示例二
a=torch.rand(1,2,3)#均匀分布(0,1)之间生成数
print(a.dim())#dimension=3
print(a.shape)
print(a.size(0))
print(a.size(1))
print(a.size(2))

#示例三
a=torch.randn(2,3)#表示正态分布，随机生成dimension=2的tensor
print(a)
print(a.dim())
print(a.shape)
print(a.size(0))
print(a.size(1))

a=torch.FloatTensor(2,3)
print(a)
print(a.dim())
print(a.shape)
print(a.size(0))
print(a.size(1))

代码运行结果

dimension表示维度

下面以图片为例

a=torch.rand(4,3,28,28)
#表示4张图片，通道数为3，RGB图像，长为28，宽为28
print(a)
print(a.shape)
print(a.dim())