Pytorch中的torch.gather函数的含义
b = torch.Tensor([[1,2,3],[4,5,6]])
print b
index_1 = torch.LongTensor([[0,1],[2,0]])
index_2 = torch.LongTensor([[0,1,1],[0,0,0]])
print torch.gather(b, dim=1, index=index_1)
print torch.gather(b, d...
转载
2019-04-18 18:13:56 ·
516 阅读 ·
0 评论