pytorch

张量

创建张量的方法

import torch
import numpy as np
t1=torch.Tensor([1,2,3])
t1
array1=np.arange(12).reshape(3,4)
array1
torch.Tensor(array1)
torch.empty(3,4)
torch.rand([3,4])
torch.randint(low=0,high=10,size=[3,4])
torch.randn([3,4])

pytorch中tensor中常用的方法

t1 = torch.Tensor([[[1]]])
t1
t1.item()`在这里插入代码片`

t2=torch.Tensor([[[1,2]]])
t2.item()

这样就会报错

t2.numpy()