qq_39522282的博客

参考官方文档https://pytorch.org/docs/stable/notes/cuda.html在如下的示例代码中：with torch.cuda.device(1): # allocates a tensor on GPU 1 a = torch.tensor([1., 2.], device=cuda) # transfers a tensor from CPU to GPU 1 b = torch.tensor([1., 2.]).cuda()

torch.gather(input, dim, index)设tt=torch.tensor([[1, 2], [3, 4]]),index=torch.tensor([[0, 0], [1, 0]])tt和index可写为tensor([[1, 2], [3, 4]])tensor([[0, 0], [1, 0]])1.当dim=1...