Функция logical_or

Функция logical_or выполняет поэлементное логическое ИЛИ над двумя тензорами. Первым и вторым параметрами передаются тензоры, которые должны иметь одинаковую форму или быть вещаемыми (broadcastable). Результатом является тензор с булевыми значениями (тип torch.bool), где каждый элемент равен True, если хотя бы один из соответствующих элементов входных тензоров истинен, и False в противном случае.

Функция также поддерживает передачу скалярного значения в качестве одного из аргументов. В этом случае скаляр приводится к тензору и операция выполняется поэлементно.

Синтаксис

torch.logical_or(input, other)

Параметры

Параметр input - первый входной тензор. Параметр other - второй входной тензор или скаляр. Оба тензора должны быть одного размера или вещаемыми.

Пример

Давайте выполним поэлементное логическое ИЛИ над двумя тензорами:

import torch

t1 = torch.tensor([True, False, True])
t2 = torch.tensor([False, False, True])
res = torch.logical_or(t1, t2)
print(res)

Результат выполнения кода:

tensor([ True, False, True])

Пример

Выполним операцию над целочисленными тензорами: ненулевые значения считаются как True, нулевые - как False:

import torch

t1 = torch.tensor([1, 0, 3])
t2 = torch.tensor([0, 2, 0])
res = torch.logical_or(t1, t2)
print(res)

Результат выполнения кода:

tensor([ True, True, True])

Пример

Использование скаляра в качестве второго аргумента:

import torch

t = torch.tensor([True, False, False])
res = torch.logical_or(t, True)
print(res)

Результат выполнения кода:

tensor([ True, True, True])

Пример

Операция с вещанием для тензоров разных форм:

import torch

t1 = torch.tensor([[True, False], [False, True]])
t2 = torch.tensor([True, False])
res = torch.logical_or(t1, t2)
print(res)

Результат выполнения кода:

tensor([
	[ True, False],
	[ True,  True],
])

Смотрите также

функцию logical_and,
которая выполняет поэлементное логическое И
функцию logical_not,
которая выполняет поэлементное логическое НЕ
функцию logical_xor,
которая выполняет поэлементное исключающее ИЛИ
функцию eq,
которая выполняет поэлементное сравнение на равенство