Pytorch
1. 创建tensor
-
使用numpy、list创建
a = torch.tensor(np.array([1,2,3,4]) b = torch.tensor([1,2,3,4])
-
使用empty、zero、ones初始化
a = torch.empty(3, 5) b = torch.zeros(2, 5) c = torch.new_ones(4, 8, 9)
-
使用随机初始化
a = torch.rand(2, 5) b = torch.randn(3, 5)
2. tensor的一些属性
-
查看tensor维度
a.ndim
-
查看tensor形状
a.shape
-
查看tensor的数据类型
a.dtype
-
查看tensor的设备
a.device
3. tensor的计算
-
数学计算
# 逐元素计算 a + b a - b a * b a / b a % b a // b # 矩阵点乘 a.dot(b) #求平均值、最大值、最小值和求和 a.mean() # 元素类型为float a.max() a.min() a.sum()
-
tensor维度变换
a.view(2,3)
-
修改tensor的设备
a.to("cuda")