Pytorch torch.ne | torch.nonzero

【Pytorch】torch.ne | torch.nonzero

• 1. torch.ne
  • 2. torch.nonzero

PyTorch张量比较函数解析

torch.ne 是 PyTorch 中一个专门用于逐元素进行不相等判断的函数，其作用是比较两个张量中对应位置的元素是否不相等。该函数会生成一个与原始输入张量具有相同维度结构的布尔类型张量，其中对应位置上元素值不同的地方会被标记为 True，而相等的位置则显示为 False。
语法

    torch.ne(input, other, *, out=None) -> Tensor
    
    
      
    

参数
input (Tensor): 需要进行对比操作的第一个输入张量。
other (Tensor 或 数值): 作为对比对象的第二个输入，既可以是张量形式，也可以是单一数值。
out (Tensor, 可选): 用于保存运算结果的输出张量。
返回
生成一个与原始输入具有相同维度结构的布尔类型张量。
代码示例:

    import torch
    
    # 创建两个张量
    tensor1 = torch.tensor([1, 2, 3, 4])
    tensor2 = torch