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])