fxolivia的专栏

>>>import torch >>> >>>a = torch.Tensor([[1,2,3],[4,5,6]]) >>>a tensor([[1., 2., 3.], [4., 5., 6.]]) >>>b = torch.gather(a,1,torch.LongTensor([[1,...

Torch 创建初始化零矩阵且数据类型为long： import torch x=torch.zeros(3, 2, dtype=torch.long) print(x) print (x.size()) 输出： tensor([[0, 0], [0, 0], [0, 0]]) torch.Size([3,2]) 直接从数据构造张量 x = t...