pytorch 不等于
时间: 2023-11-07 20:03:00 浏览: 33
pytorch与lightning 使用的不同
在Pytorch中,判断两个Tensor是否不相等可以使用torch.ne()函数。该函数会逐元素比较两个Tensor,并返回一个新的Tensor,其中每个元素为对应位置元素是否不相等的布尔值。
例子:
import torch
a = torch.Tensor([1,2,3])
b = torch.Tensor([1,2,4])
print(torch.ne(a,b))
输出结果:
tensor([False, False, True])
阅读全文