torch.gather 理解
torch.gather 理解查看如下代码b = torch.Tensor([[1,2,3],[4,5,6]])print bindex_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)p...
原创
2020-04-03 10:34:23 ·
198 阅读 ·
0 评论