PyTorch系列2 ---Tensor的操作

## Tensor操作中的api和NumPy非常相似，，基本是一致的
x = torch.ones(2,2)
print("x: ",x) #这是一个float tensor
print("x.type(): ",x.type())
## 将其转化为整型
x = x.long()  #等价于 x = x.type(torch.LongTensor)
print("转化为整型的x: ",x)
print("转化为整型的x的type: ",x.type())
#再将其转回 float
x = x.float() # 等价于 x = x.type(torch.FloatTensor)
print("转换为float的x: ",x)
print("转换为float的x的type: ",x.type())

运行结果：

x = torch.randn(4,3)
print("x的值： ",x)
#沿着行取最大值
max_value, max_idx = torch.max(x,dim=1)
print("每一行的最大值：",max_value)
print("每一行的最大值下标：",max_idx)
print("每一行的最大值：","每一行的最大值下标：")
#沿着行对x 求和
sum_x = torch.sum(x,dim=1)
print("沿着行对x 求和: ",sum_x)

运行结果：