本文简要介绍python语言中 torch.allclose
的用法。
用法:
torch.allclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False) → bool
此函数检查是否所有
input
和other
都满足条件:元素层面,对于
input
和other
的所有元素。该函数的行为类似于numpy.allclose例子:
>>> torch.allclose(torch.tensor([10000., 1e-07]), torch.tensor([10000.1, 1e-08])) False >>> torch.allclose(torch.tensor([10000., 1e-08]), torch.tensor([10000.1, 1e-09])) True >>> torch.allclose(torch.tensor([1.0, float('nan')]), torch.tensor([1.0, float('nan')])) False >>> torch.allclose(torch.tensor([1.0, float('nan')]), torch.tensor([1.0, float('nan')]), equal_nan=True) True
参数:
相关用法
- Python PyTorch all_reduce用法及代码示例
- Python PyTorch all_gather用法及代码示例
- Python PyTorch allreduce_hook用法及代码示例
- Python PyTorch all_to_all用法及代码示例
- Python PyTorch all用法及代码示例
- Python PyTorch all_gather_object用法及代码示例
- Python PyTorch argsort用法及代码示例
- Python PyTorch addmm用法及代码示例
- Python PyTorch addmv用法及代码示例
- Python PyTorch apply_effects_tensor用法及代码示例
- Python PyTorch assert_close用法及代码示例
- Python PyTorch angle用法及代码示例
- Python PyTorch atanh用法及代码示例
- Python PyTorch annotate用法及代码示例
- Python PyTorch async_execution用法及代码示例
- Python PyTorch argmax用法及代码示例
- Python PyTorch atan用法及代码示例
- Python PyTorch as_strided用法及代码示例
- Python PyTorch acos用法及代码示例
- Python PyTorch avg_pool1d用法及代码示例
- Python PyTorch asin用法及代码示例
- Python PyTorch argmin用法及代码示例
- Python PyTorch any用法及代码示例
- Python PyTorch asinh用法及代码示例
- Python PyTorch add用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.allclose。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。