RuntimeError: Expected object of scalar type Float but got scalar type Int for argument #2 'other'

不同类型的是不相等的,并且无法比较

import torch
a=torch.tensor([1],dtype=torch.float)
b=torch.tensor([1],dtype=torch.int)
#tensor 的1无论什么情况都与普通的1相等
#不同类型的是不相等的,并且无法比较
if a==b:
    print((a==b))
    print((a==b).long().dtype)
else:
    print("not equal")

结果是错误

    if a==b:
RuntimeError: Expected object of scalar type Float but got scalar type Int for argument #2 'other'

默认的1和tensor中各种类型都相等

import torch
a=torch.tensor([1],dtype=torch.float)
#b=torch.tensor([1],dtype=torch.int)
b=1
#tensor 的1无论什么情况都与普通的一相等
#不同类型的是不相等的,并且无法比较
if a==b:
    print((a==b))
    print((a==b).long().dtype)
else:
    print("not equal")

结果

tensor([1], dtype=torch.uint8)
torch.int64

 

你可能感兴趣的:(python,pytorch)