[ 仅 paddle 参数更多 ]torch.argwhere

torch.argwhere

torch.argwhere(input)

paddle.nonzero

paddle.nonzero(x, as_tuple=False)

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

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | x | 输入的 Tensor ,仅参数名不一致。 | | - | as_tuple | 返回格式。是否以 1-D Tensor 构成的元组格式返回。 Pytorch 无此参数, Paddle 保持默认即可。 |