torch使用

1.张量的创建方法

torch.tensor(list)

torch.empty()/zeros()/ones()

torch.rand()/randint()/randn()

2.张量的属性和方法

数据中只有一个元素的时候 tensor.item()

转化为numpy数组 tensor.numpy()

获取形状 tensor.size()     获取第一个维度的形状  tensor.size(1)

形状的变化  tensor.view([shape])

获取维度tensor.dim()

常用计算方法 tensor.max()/min()/std

转置 ：二维tensor.t()  tensor.transpose(0,1)

            高维tensor.permute(0,2,1)

3.数据类型：torch.tensor（array.dtype）torch.ones(dtype)

获取数据类型  tensor.dtype

修改数据类型： tensor.float()/long()/int()

4.原地修改的方法：x.add_(y)