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