Función PyTorch torch.any

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.anyEs una función en PyTorch que se utiliza para determinar si hay algún elemento que sea True.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

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

print("Hay True:", torch.any(x))
print("dim=0 hay True:", torch.any(x, dim=0))

El resultado de salida es:

有 True: tensor(True)
dim=0 有 True: tensor([True, False, False])

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones