[仅 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 保持默认即可。