[ 仅 paddle 参数更多 ]torch.Tensor.argwhere

torch.Tensor.argwhere

torch.Tensor.argwhere()

paddle.Tensor.nonzero

paddle.Tensor.nonzero(as_tuple=False)

其中 Paddle 相比 PyTorch 支持更多其他参数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
- as_tuple 返回格式。是否以 1-D Tensor 构成的元组格式返回。 PyTorch 无此参数, Paddle 保持默认即可。