python - テンソルの名前を付けてテンソル インデックスを取得する

Question

テンソルの特定のコンポーネントから配列を作成したいと思います。すばらしいコマンドを見つけましたnp.argwhere()。これは、特定の基準を満たすテンソルのインデックスを返しますが、テンソルのコンポーネントとして名前を付けません。つまり、テンソルの対として返され[0,0,1,1]ます。 x[0,0,1,1]x

コンポーネントがインデックスと添付されたテンソルの名前で書かれている特定の基準を満たすテンソルのコンポーネントを取得するための組み込みまたは巧妙な方法はありますか?

Answer 1

where代わりに使用できますargwhere...

>>> x = np.arange(6).reshape(2,3)
>>> x
array([[0, 1, 2],
       [3, 4, 5]])
>>> np.argwhere(x > 1)
array([[0, 2],
       [1, 0],
       [1, 1],
       [1, 2]])
>>> np.where(x > 1)
(array([0, 1, 1, 1]), array([2, 0, 1, 2]))
>>> x[np.where(x > 1)]
array([2, 3, 4, 5])