检查PyTorch张量是否在给定的epsilon误差内相等

Question

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

由于浮点误差，我想知道元素是否仅以小的epsilon值不同。

- Tom Hale

Answer 1

目前稳定版本（0.4.1）中并未记录此函数，但可以在“master (unstable)”分支中找到相关文档。

torch.allclose()函数将返回一个布尔值，指示所有元素差异是否相等，可容忍一定误差。

另外还有一个未记录的isclose()函数：

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

- Tom Hale

---

现在它已经稳定了。 - Mike W

希望能有一些函数来基于张量范数测试张量的相似度，这将非常方便。 - a06e