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 ...
確定! 回上一頁