Функция torch.nonzero в PyTorch

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.nonzeroэто функция в PyTorch, используемая для возврата индексов ненулевых элементов.

Определение функции

torch.nonzero(input)

Примеры использования

Пример

import torch

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

# Возвращаем индексы ненулевых элементов
индексы = torch.nonzero(x)

print("Индексы ненулевых элементов:")
print(indices)

Результат вывода:

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

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения