本文简要介绍python语言中 torch.count_nonzero
的用法。
用法:
torch.count_nonzero(input, dim=None) → Tensor
沿着给定的
dim
计算张量input
中非零值的数量。如果未指定dim,则对张量中的所有非零值进行计数。例子:
>>> x = torch.zeros(3,3) >>> x[torch.randn(3,3) > 0.5] = 1 >>> x tensor([[0., 1., 1.], [0., 0., 0.], [0., 0., 1.]]) >>> torch.count_nonzero(x) tensor(3) >>> torch.count_nonzero(x, dim=0) tensor([0, 1, 2])
相关用法
- Python PyTorch column_stack用法及代码示例
- Python PyTorch cov用法及代码示例
- Python PyTorch cos用法及代码示例
- Python PyTorch compute_deltas用法及代码示例
- Python PyTorch conv_transpose3d用法及代码示例
- Python PyTorch combinations用法及代码示例
- Python PyTorch conv2d用法及代码示例
- Python PyTorch collect_all用法及代码示例
- Python PyTorch convert用法及代码示例
- Python PyTorch conv1d用法及代码示例
- Python PyTorch constant_用法及代码示例
- Python PyTorch context用法及代码示例
- Python PyTorch cosh用法及代码示例
- Python PyTorch cosine_similarity用法及代码示例
- Python PyTorch conv3d用法及代码示例
- Python PyTorch cond用法及代码示例
- Python PyTorch conj_physical用法及代码示例
- Python PyTorch conj用法及代码示例
- Python PyTorch complex用法及代码示例
- Python PyTorch corrcoef用法及代码示例
- Python PyTorch copysign用法及代码示例
- Python PyTorch conv_transpose2d用法及代码示例
- Python PyTorch conv_transpose1d用法及代码示例
- Python PyTorch cholesky用法及代码示例
- Python PyTorch cumprod用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.count_nonzero。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。