torch.Tensor.count_nonzero

count_nonzero(dim=None) → 张量

参见 torch.count_nonzero()