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

torch.argwhere ¶

          torch.argwhere(input)

         

          paddle.nonzero(x, as_tuple=False)

         

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

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