gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-03 16:13:16 31 4
gpt4 key购买 nike

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

给定浮点错误,我想知道元素是否仅以较小的ε值不同。

最佳答案

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

torch.allclose() 将返回一个 bool 值,该 bool 值指示所有逐元素差异是否相等,从而允许出现误差。

此外,还有未记录的isclose():

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

关于pytorch - 检查PyTorch张量在epsilon内是否相等,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53374928/

31 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com