快捷方式

torch.ne

torch.ne(input, other, *, out=None) 张量

逐元素计算 inputother\text{input} \neq \text{other}

第二个参数可以是一个数或形状与第一个参数可 广播 的张量。

参数
关键字参数

out (张量, 可选) – 输出张量。

返回

一个布尔张量,在 input 不等于 other 的地方为 True,在其他地方为 False

示例

>>> torch.ne(torch.tensor([[1, 2], [3, 4]]), torch.tensor([[1, 1], [4, 4]]))
tensor([[False, True], [True, False]])

文档

查阅 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源