Pytorch 之gather的用法 · input (Tensor) – The source tensor · dim (int) – The axis along which to index · index (LongTensor) – The indices of elements to gather.
確定! 回上一頁