Función PyTorch torch.nonzero

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

torch.nonzeroEs una función en PyTorch que se utiliza para devolver los índices de los elementos distintos de cero.

Definición de la función

torch.nonzero(input)

Ejemplo de uso

Ejemplo

import torch

x = torch.tensor([[0, 1], [2, 0], [0, 3]])

# Devuelve los índices de los elementos distintos de cero
indices = torch.nonzero(x)

print("Índices de elementos distintos de cero:")
print(indices)

El resultado de salida es:

非零元素索引:
tensor([[0, 1],
        [1, 0],
        [2, 1]])

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

Otras extensiones