Fonction torch.all de PyTorch

Pytorch torch 参考手册 Manuel de référence Pytorch torch

torch.allC'est une fonction de PyTorch utilisée pour déterminer si tous les éléments sont True.

Définition de fonction

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

Exemples d'utilisation

Exemple

import torch

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

# Vérification globale
print("Tous sont True :", torch.all(x))

# Vérification selon la dimension
print("dim=0 tous sont True :", torch.all(x, dim=0))

La sortie est :

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

Pytorch torch 参考手册 Manuel de référence Pytorch torch

Autres extensions