[仅 Paddle 参数更多]torch.Tensor.count_nonzero¶

torch.Tensor.count_nonzero¶

torch.Tensor.count_nonzero(dim=None)

paddle.Tensor.count_nonzero¶

paddle.Tensor.count_nonzero(axis=None, keepdim=False, name=None)

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

参数映射¶

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | dim | axis | 表示进行运算的轴，仅参数名不一致。 | | - | keepdim | 是否在输出 Tensor 中保留减小的维度， Pytorch 无此参数， Paddle 保持默认即可。 |