batchサイズx max_lenxの配列があるとしましょうoutput_size。ここbatchで、、、max_lenおよびoutput_sizeすべてが正の自然数に対応します。次元 1 (つまりmax_len) の個々の項目に対応するインデックスのリストがあります。これらのインデックスを指定して配列から選択するにはどうすればよいですか?
具体的な例として、次のようなものがあるとします。
>>> l = np.random.randn(4,5,6)
>>> l.shape
(4, 5, 6)
>>> idx = [0,0,2,3]
l与えられたものを選択すると、次のidxようになります。
>>> l[:,idx,:].shape
(4, 4, 6)
>>>
私も試しnp.takeましたが、同じ結果に達しました：
>>> np.take(l,idx,axis=1).shape
(4, 4, 6)
>>>
ただし、私が見ている出力は、 (つまり、最初の次元)(4,1,6)の各要素を 1 つのアイテムだけで見ようとしているからです。batch適切な形状で出力を生成するにはどうすればよいですか?