[ torch 参数更多 ]torch.nonzero¶
torch.nonzero¶
torch.nonzero(input,
*,
out=None,
as_tuple=False)
参数映射¶
| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | x | 表示输入的 Tensor ,仅参数名不一致。 | | out | - | 表示输出的 Tensor , Paddle 无此参数,需要转写。 | | as_tuple | as_tuple | bool 类型表示输出数据的格式,默认 False 时,输出一个张量,True 时输出一组一维张量。 |