torch.all

torch.all(input) → Tensor

Проверяет, что все элементы в input равны True.

Примечание

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

Пример:

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

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

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

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

Параметры:

• input (Тензор) – входной тензор.
• dim (целое) – измерение для уменьшения.
• keepdim (булево) – сохраняется ли выходной размер.

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

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

Пример:

>>> a = torch.rand(4, 2).bool()
>>> a
tensor([[True, True],
        [True, False],
        [True, True],
        [True, True]], dtype=torch.bool)
>>> torch.all(a, dim=1)
tensor([ True, False,  True,  True], dtype=torch.bool)
>>> torch.all(a, dim=0)
tensor([ True, False], dtype=torch.bool)