[ 仅 paddle 参数更多 ] torch.Tensor.take¶
torch.Tensor.take¶
torch.Tensor.take(indices)
paddle.Tensor.take¶
paddle.Tensor.take(index, mode='raise', name=None)
两者功能一致，仅参数名不一致，其中 Paddle 相比 PyTorch 支持更多其他参数,具体如下：
参数映射¶
| PyTorch | PaddlePaddle | 备注 |
|---|---|---|
| indices | index | 表示输入 tensor 的索引，仅参数名不一致。 |
| - | mode | 指定索引越界的 3 种处理方式，PyTorch 无此参数，Paddle 保持默认即可。 |
注：
三种 mode
mode='raise'，若索引越界，通过最后调用的paddle.index_select抛出错误 （默认）；mode='wrap'，通过取余约束越界的 indices；mode='clip'，通过paddle.clip将两端超出范围的索引约束到 [0, max_index-1]。