torch.gather()
torch.gather()
torch中API如下讲解:
torch.gather(input, dim, index, out=None, sparse_grad=False) → Tensor
Gathers values along an axis specified by dim. For a 3-D tensor the output is specified by:
out[i...
原创
2019-07-22 15:16:36 ·
1086 阅读 ·
0 评论