- 基本数据类型
python:Int, float, Int array, Float array, string
pytorch: IntTensor of size, FloatTensor of size, IntTensor of size [d1,,,dn], FloatTensor of size[d1,,,dn], (one-hot,Embedding)
重点是(CPU):FloatTensor, ByteTensor, IntTensor, LongTensor
(GPU): 数据类型不同,中间增加了cuda
常用的语法:a = torch.randn(2,3) 随机生成两行三列
a.type() a的数据类型
isinstance(a,b) 数据类型是否匹配
- tensor
dimension 0 :标量 比如 损失 a = torch.tensor(2.2) a.shape = torch.size([])
1 :向量 a = torch.tensor([1.1]) (多了一个中括号)
2 :矩阵 a = torch.rand(2,3) (一维,加数量)
3 :立方体 a.torch.rand(2,2,3) (二维,加数量)
4 : 适合图片(3维,加数量)