2
TensorFlowで選択した軸に沿ってテンソルの最大要素のインデックスを取得するにはどうすればよいですか?TensorFlowテンソルの最大要素のインデックス
TensorFlowで選択した軸に沿ってテンソルの最大要素のインデックスを取得するにはどうすればよいですか?TensorFlowテンソルの最大要素のインデックス
申し訳ありませんが、すでにそれが分かりました。これは、次のとおりです。
tf.argmax(input, dimension, name=None)