检查PyTorch张量在epsilon内是否相等

Question

如何检查两个PyTorch张量在语义上是否相等？

鉴于浮点错误，我想知道这些元素是否仅以较小的ε值不同。

Answer 1

在撰写本文时，这是最新稳定版本（0.4.1）中尚未记录的功能，但是该文档位于master (unstable)分支中。

torch.allclose()将返回一个布尔值，指示所有逐元素差异是否相等，以允许出现误差。

此外，还有未记录的isclose()：

>>> torch.isclose(torch.Tensor([1]), torch.Tensor([1.00000001]))
tensor([1], dtype=torch.uint8)