如何检查两个 PyTorch 张量在语义上是否相等？
考虑到浮点错误，我想知道元素是否仅相差一个小的 epsilon 值。
在撰写本文时，这是最新稳定版本 (0.4.1) 中的一个未记录的函数，但文档位于master (unstable) branch.
torch.allclose() https://pytorch.org/docs/master/torch.html?highlight=allclose#torch.allclose将返回一个布尔值，指示所有元素差异是否相等，并允许一定的误差范围。
此外，还有无证的isclose():
>>> torch.isclose(torch.Tensor([1]), torch.Tensor([1.00000001]))
tensor([1], dtype=torch.uint8)
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
