テンソルをテンソルフローのインデックスにする方法

Question

2Dテンソルの形状（サイズ、サイズ）を持っていて、元のテンソルの行インデックスと列インデックスを含む2つの新しいテンソルを取得したいとします。 だから、サイズが2であれば、私は 取得したい[[0、0]、[1、1]]と[[0、1]、[0、1]]

何トリッキーだが、そのサイズが他のですテンソルは、テンソルフローセッションでグラフを実行するときにのみ値を知ることができます。 テンソルフローでこれを行うにはどうすればよいですか？

Answer 1

あなたはtf.meshgridをお探しのようです。ここ

は、例えば次のとおり

shape = tf.shape(matrix) 
R, C = tf.meshgrid(tf.range(shape[0]), tf.range(shape[1]), indexing='ij') 

matrixはあなたの2次元テンソル、R及びCは、それぞれ、自分の行と列のインデックスを含んでなります。行列が正方形の場合（わずか1つのtf.range）、これはわずかに単純化することができます。