linalg:take-rows

(linalg:take-rows array indices)

インデックスベクタ indices で選んだ axis-0 のスライスを、入力と同じ幅の新しい配列として返します (numpy の x[mask] / np.take(a, idx, axis=0))。スライス (slab) を行優先で丸ごとコピーするため、rank >= 1 の任意の rank で動作します。インデックス値は整数に truncate され、同じインデックスを重複して指定できます。linalg:choice や linalg:permutation の結果と組み合わせるとミニバッチ抽出になります。