import torch a = torch.arange(3, 12).view(3, 3) print(a) index = torch.tensor([[1, 1]]) print(torch.gather(a, dim=0, index=index)). Python.
確定! 回上一頁