PyTorch torch.nonzero 関数
Pytorch torch リファレンスマニュアルtorch.nonzeroPyTorch で非ゼロ要素のインデックスを返すために使用される関数です。
関数定義
torch.nonzero(input)
使用例
実例
import torch
x = torch.tensor([[0, 1], [2, 0], [0, 3]])
# 非ゼロ要素のインデックスを返す
indices = torch.nonzero(x)
print("非ゼロ要素のインデックス:")
print(indices)
x = torch.tensor([[0, 1], [2, 0], [0, 3]])
# 非ゼロ要素のインデックスを返す
indices = torch.nonzero(x)
print("非ゼロ要素のインデックス:")
print(indices)
出力結果は次のとおりです：
非零元素索引:
tensor([[0, 1],
[1, 0],
[2, 1]])
その他の拡張