torch.gather函数声明为:torch.gather(input, dim, index, out=None, sparse_grad=False) → Tensor函数作用为: Gathers values along an axis ...
確定! 回上一頁