Функция PyTorch torch.all

Pytorch torch 参考手册 Справочное руководство Pytorch torch

torch.allЭто функция в PyTorch, используемая для определения, все ли элементы равны True.

Определение функции

torch.all(input, dim, keepdim, out)

Примеры использования

Пример

import torch

x = torch.tensor([[True, True, True], [True, False, True]])

# Глобальная проверка
print("Все равны True:", torch.all(x))

# Проверка по измерению
print("dim=0 все равны True:", torch.all(x, dim=0))

Вывод:

全部为 True: tensor(False)
dim=0 全部为 True: tensor([True, False, True])

Pytorch torch 参考手册 Справочное руководство Pytorch torch

Другие расширения