torch.gather()torch.gather(input, dim, index, *, sparse_grad=False, out=None) → Tensor参数解释:input (Tensor) – the source tensordim (int) – the axis ...
確定! 回上一頁