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) → Tensor

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

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

Параметры

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

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

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

Пример:

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