Función PyTorch torch.is_nonzero

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

torch.is_nonzeroEs una función en PyTorch que se utiliza para comprobar si un tensor es un único elemento no nulo.

Definición de la función

torch.is_nonzero(input)

Ejemplo de uso

Ejemplo

import torch

print(torch.is_nonzero(torch.tensor([1.0])))
print(torch.is_nonzero(torch.tensor([0.0])))
print(torch.is_nonzero(torch.tensor([1, 2])))

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

Otras extensiones