Pytorch学习笔记（一）——数据&&运算

一、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) # 矩阵除法