2. Max'i bir Tensorflow tensöründen nasıl alabilirim? Ben aşağıdaki kullanarak denedi</p> <pre><code>p = tf.stop_gradient(tf.argmax(prev, 1)) </code></pre> <p>ama dimn uyumsuzdur:

Question

Şu anda, benim işlevi argmax kullanan tek vardır top_k neden olan

p = tf.stop_gradient(tf.nn.top_k(prev, 2)[1]) 

raise ValueError("Linear is expecting 2D arguments: %s" % str(shapes)) 
ValueError: Linear is expecting 2D arguments: [[None, 2, 1024], [None, 1024]] 

Benim TF versiyonu 0.5 olabilir 2 adet

Answer 1

tf.nn.top_k() için belgelere bakın. Fonksiyon, değerler ve indeksler döndürür. Yani aşağıdaki gibi bir şey çalışmalıdır.

values, indices = tf.nn.top_k(prev,2) 
p = tf.stop_gradient(indices[1])