torch.equal函数怎么使用

本篇内容介绍了“torch.equal函数怎么使用”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!

代码实验:

(base) PS C:\Users\chenxuqi> python
Python 3.7.4 (default, Aug  9 2019, 18:34:13) [MSC v.1915 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.>>> import torch>>> torch.equal(torch.tensor([1, 2]), torch.tensor([1, 2]))True>>> torch.equal(torch.tensor([1, 2]), torch.tensor([[1],[2]]))False>>> torch.equal(torch.tensor([[1, 2]]), torch.tensor([[1],[2]]))False>>>>>> torch.equal(torch.tensor([1, 2]), torch.tensor([1, 2.0]))Traceback (most recent call last):
  File "<stdin>", line 1, in <module>RuntimeError: Expected object of scalar type Long but got scalar type Float for argument #2 'other'>>> torch.equal(torch.tensor([1, 2]), torch.tensor([1.0, 2.0]))Traceback (most recent call last):
  File "<stdin>", line 1, in <module>RuntimeError: Expected object of scalar type Long but got scalar type Float for argument #2 'other'>>> torch.equal(torch.tensor([1.0, 2.0]), torch.tensor([1.0, 2.0]))True>>>>>>

“torch.equal函数怎么使用”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识可以关注网站,小编将为大家输出更多高质量的实用文章!