pytorch文档学习
(1)
torch.gather(input, dim, index, out=None) → Tensor
input是输入tensor;dim维度(二维矩阵,0表示在列的方向,行的索引;1表示在行的方向,列的索引);index索引
out[i][j][k] = input[index[i][j][k]][j][k] # dim=0
out[i][j][k] = input[i][index...
原创
2019-01-04 11:29:49 ·
208 阅读 ·
0 评论