比较与torch.logical_and的功能差异

查看源文件

torch.logical_and

class torch.logical_and(input, other, out=None)

更多内容详见 torch.logical_and

mindspore.ops.LogicalAnd

class class mindspore.ops.LogicalAnd()(x, y)

更多内容详见 mindspore.ops.LogicalAnd

使用方式

PyTorch: 计算给定输入张量的逐元素逻辑与。零被视为“False”,非零被视为“True”。

MindSpore: 按元素计算两个输入张量的逻辑与。输入可以是bool值或数据类型为bool的张量。

代码示例

import numpy as np
import torch
import mindspore.ops as ops
import mindspore as ms

# MindSpore
x = ms.Tensor(np.array([True, False, True]), ms.bool_)
y = ms.Tensor(np.array([True, True, False]), ms.bool_)
logical_and = ops.LogicalAnd()
print(logical_and(x, y))
# [ True False False]
x = ms.Tensor(np.array([True, False, True]), ms.int32)
y = ms.Tensor(np.array([True, True, False]), ms.bool_)
logical_and = ops.LogicalAnd()
print(logical_and(x, y))
# TypeError: For primitive[LogicalAnd], the input argument[x] must be a type of {Tensor[Bool],}, but got Int32.

# Pytorch
print(torch.logical_and(torch.tensor([True, False, True]), torch.tensor([True, False, False])))
# tensor([ True, False, False])
a = torch.tensor([0, 1, 10, 0], dtype=torch.int8)
b = torch.tensor([4, 0, 1, 0], dtype=torch.int8)
print(torch.logical_and(a, b))
# tensor([False, False,  True, False])
print(torch.logical_and(a.double(), b.double()))
# tensor([False, False,  True, False])
print(torch.logical_and(a.double(), b))
# tensor([False, False,  True, False])
print(torch.logical_and(a, b, out=torch.empty(4, dtype=torch.bool)))
# tensor([False, False,  True, False])