torch.any

torch.any(input) → Tensor

Проверяет, если какой-либо элемент в input принимает значение True.

Примечание

Эта функция соответствует поведению NumPy, возвращая выходные данные типа bool для всех поддерживаемых типов данных, кроме uint8. Для uint8 тип данных выходных данных — uint8 сам.

Пример:

>>> a = torch.rand(1, 2).bool()
>>> a
tensor([[False, True]], dtype=torch.bool)
>>> torch.any(a)
tensor(True, dtype=torch.bool)
>>> a = torch.arange(0, 3)
>>> a
tensor([0, 1, 2])
>>> torch.any(a)
tensor(True)

torch.any(input, dim, keepdim=False, *, out=None) → Тензор

Для каждой строки input в заданном измерении dim, возвращает True , если какой-либо элемент в строке принимает значение True, и False в противном случае.

Если keepdim равно True, тензор вывода имеет тот же размер, что и input, за исключением измерения dim, где он имеет размер 1. В противном случае, dim сжимается (см. torch.squeeze()), что приводит к тому, что тензор вывода имеет на одно измерение меньше, чем input.

Параметры

• input (Тензор) – входной тензор.
• dim (int) – измерение для сокращения.
• keepdim (bool) – сохранять ли размерность вывода dim или нет.

Ключевые аргументы

out (Тензор, необязательно) – тензор вывода.

Пример:

>>> a = torch.randn(4, 2) < 0
>>> a
tensor([[ True,  True],
        [False,  True],
        [ True,  True],
        [False, False]])
>>> torch.any(a, 1)
tensor([ True,  True,  True, False])
>>> torch.any(a, 0)
tensor([True, True])