[ paddle 参数更多 ]torch.Tensor.count_nonzero

torch.Tensor.count_nonzero 

          torch.Tensor.count_nonzero(dim=None)

         

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

         

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

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