3次元配列でargmaxテンソルフロー関数を使うには？

Question

tf.argmaxの3D配列の使い方を知りたい。

私の入力データがそのようなものです：

[[[0, -1, 5, 2, 1], [2, 2, 3, 2, 5], [6, 1, 2, 4, -1]], 
[[-1, -2, 3, 2, 1], [0, 3, 2, 7, -1], [-1, 5, 2, 1, 3]]] 

そして、私はこのように、この入力データによってARGMAXの出力を取得したい：

[[2, 4, 0], [2, 3, 1]] 

そして、私はこの中でsoftmax_cross_entropy_with_logits機能を使用したいですフォーマット。

tf.nn.softmax_cross_entropy_with_logitsの機能とtf.equal(tf.argmax)とtf.reduce_mean(tf.cast)の使用方法を教えてください。

Answer 1

あなたはaxis=3

a = tf.constant([[[0, -1, 5, 2, 1], [2, 2, 3, 2, 5], [6, 1, 2, 4, -1]], 
     [[-1, -2, 3, 2, 1], [0, 3, 2, 7, -1], [-1, 5, 2, 1, 3]]]) 
b = tf.argmax(a, axis=2) 

沿っ tf.argmaxを使用することができます