- 博客(1)
- 收藏
- 关注
原创 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 1088
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人