pytorch学习笔记2:比较操作
目录 torch.eq torch.equal torch.ge torch.gt torch.eq torch.eq(input, other, out=None) → Tensor 比较 元素相等性 。第二个参数可为一个数或与第一个参数同类型形状的张量。 参数: input (Tensor) – 待比较张量 other (Tensor or float) – 比较张量或数 out (Tensor, optional) – 输出张量,须为 ByteTensor类型 or 与 input 同类型 返回值: 一个 torch.ByteTensor 张量,包含了每个位置的比较结果(相等为1,不等为0 ) 返回类型: Tensor 例子: >>> torch.eq(torch.Tensor([[1, 2], [3, 4]]), torch.Tensor([[1, 1], [4, 4]])) 1 0 0 1 [torch.ByteTensor of size 2x2] torch.equal torch.equal(tensor1, tensor2) → bool 如果 两个张量有相同的形状和元素值,则返回 True ,否则 False 。 例子: >>> torch.equal(torch.Tensor([1, 2]), torch.Tensor([1, 2])) True >>>a