一、Tensor的数据类型
1.1 torch.FloatTensor
torch.FloatTensor(2, 3) # 随机生成
torch.FloatTensor([2, 3, 4, 5])
1.2 torch.IntTensor
torch.IntTensor(2, 3) # 随机生成
torch.IntTensor([2, 3, 4, 5])
1.3 torch.rand
torch.rand(2, 3) # 0~1均匀分布浮点数据
1.4 torch.randn
torch.rand(2, 3) # 0~1正态分布(均值0,方差1)浮点数据
1.5 torch.range
torch.range(1, 20, 1) # 起始值、范围结束值、步长(左闭右闭)
1.6 torch.zeros
torch.zeros(2, 3) # 浮点型0
二、Tensor的数据运算
2.1 torch.abs
2.2 torch.add
2.3 torch.clamp
torch.clamp(tensor, 1, -1) # 需要裁剪的Tensor、裁剪的上边界、裁剪的下边界
2.4 torch.div
2.5 torch.mul
2.6 torch.pow
2.7 torch.mm
torch.mm(a, b) # 矩阵乘法
2.8 torch.mv
torch.mv(a, b) # 矩阵除法