有没有办法在两个不同的python程序中比较两个pytorch Tensor对象?

cbwuti44  于 2023-03-11  发布在  Python
关注(0)|答案(1)|浏览(126)

我正在调试两个Python程序,它们都有一个变量img(具体类型是torch.Tensor),现在我需要确定这两个img是否二进制一致,因为它们在不同的程序中,我无法直接使用调试器比较它们。
我找到的唯一方法是将它们转储到本地文件中,然后加载到同一个程序中进行比较,或者使用beyondcompare等工具进行比较,但我觉得这种方法太不方便了。
有没有像md5sum这样的方法,即只根据内存中变量的二进制信息生成唯一标识符(这样我就可以直观地比较这两个标识符,因为它们在不同的程序中)?或者其他更方便的方法?

iibxawm4

iibxawm41#

我正在基于this编写我的响应
假设您的两个程序文件在同一个包中,您可以尝试将其中一个程序文件导入到另一个程序文件中,方法是添加import programName,其中programName是另一个程序的名称,但末尾不带.py。
然后,您可以通过执行类似imgOther = programName.img的操作来声明一个变量,该变量将存储来自其他程序的img变量
现在,我认为您应该能够比较imgimgOther,并实现您的目标。

相关问题