1
2DテンソルAが浮動小数点で、1DテンソルBがintであるとします。 Bの数字はA.の行のインデックスを表します。テンソルフローグラフ内でこのインデックスの検索を効率的に実行するにはどうすればよいですか?2Dテンソルのインデックス行
2DテンソルAが浮動小数点で、1DテンソルBがintであるとします。 Bの数字はA.の行のインデックスを表します。テンソルフローグラフ内でこのインデックスの検索を効率的に実行するにはどうすればよいですか?2Dテンソルのインデックス行
すでにtf.gather()を試しましたか?
も、この質問を参照してください:TensorFlow: using a tensor to index another tensor
んが、私はしませんでした。ありがとう。このコードは 'tf.diag_part(tf.gather(A、B、axis = 1))' –