Función torch.all de PyTorch
Manual de referencia de Pytorch torch
torch.allEs una función de PyTorch utilizada para determinar si todos los elementos son True.
Definición de la función
torch.all(input, dim, keepdim, out)
Ejemplo de uso
Ejemplo
import torch
x = torch.tensor([[True, True, True], [True, False, True]])
# Determinación global
print("Todos son True:", torch.all(x))
# Determinación a lo largo de una dimensión
print("dim=0 todos son True:", torch.all(x, dim=0))
x = torch.tensor([[True, True, True], [True, False, True]])
# Determinación global
print("Todos son True:", torch.all(x))
# Determinación a lo largo de una dimensión
print("dim=0 todos son True:", torch.all(x, dim=0))
El resultado de salida es:
全部为 True: tensor(False) dim=0 全部为 True: tensor([True, False, True])
Otras extensiones