PyTorch torch.logical_or 函数
torch.logical_or 是 PyTorch 中用于计算两个张量逻辑或(OR)运算的函数。
函数定义
torch.logical_or(input, other, out=None)
使用示例
实例
import torch
# 创建两个布尔张量
x = torch.tensor([True, False, True, False])
y = torch.tensor([True, True, False, False])
# 逻辑或运算
z = torch.logical_or(x, y)
print("logical_or:", z)
# 创建两个布尔张量
x = torch.tensor([True, False, True, False])
y = torch.tensor([True, True, False, False])
# 逻辑或运算
z = torch.logical_or(x, y)
print("logical_or:", z)
输出结果为:
logical_or: tensor([True, True, True, False])

Pytorch torch 参考手册