テンソルを分割する

Question

Tをテンソル形状とすると、バッチを表す[n,f]となります。今私は、axis=0に沿ってTをmテンソルにスライスしたいと思います。 mの値は、現在のバッチによって異なります。私は、形状が[m,2]の別のテンソルIを持っています。このテンソルは、スライスの発生場所を示すインデックスのペアを格納しています。

tf.sliceのインデックスを「反復」する方法がわかりません。何か案は？

これはどういうわけかtf.scanを使って実現できますか？

Answer 1

split機能を探しているとします。