不同类型的是不相等的,并且无法比较
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