torch.gather(input, dim, index, *, sparse_grad=False, out=None) → Tensor. 沿 dim 所指定的轴收集值。 对于3-D张量来说,输出由以下方式指定。
確定! 回上一頁