[仅 paddle 参数更多]torch.count_nonzero¶
torch.count_nonzero¶
torch.count_nonzero(input, dim=None)
paddle.count_nonzero¶
paddle.count_nonzero(x,
axis=None,
keepdim=False,
name=None)
其中 Paddle 相比 PyTorch 支持更多其他参数,具体如下:
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
input | x | 表示输入的 Tensor ,仅参数名不一致。 |
dim | axis | 表示进行运算的轴,仅参数名不一致。 |
- | keepdim | 是否在输出 Tensor 中保留减小的维度, PyTorch 无此参数, Paddle 保持默认即可。 |