Как работает Argmax, когда дается 3D -тензор - Tensorflow ⇐ Python

[Anonymous]

Мне было интересно, как работает Argmax, когда дается 3D -тензор. Я знаю, что происходит, когда у него есть 2D Tesnor, но 3D меня сильно сбивает с толку. < /P>
Пример: < /p>
import tensorflow as tf
import numpy as np

sess = tf.Session()

coordinates = np.random.randint(0, 100, size=(3, 3, 2))
coordinates
Out[20]:
array([[[15, 23],
[ 3, 1],
[80, 56]],
[[98, 95],
[97, 82],
[10, 37]],
[[65, 32],
[25, 39],
[54, 68]]])
sess.run([tf.argmax(coordinates, axis=1)])
Out[21]:
[array([[2, 2],
[0, 0],
[0, 2]], dtype=int64)]



Подробнее здесь: https://stackoverflow.com/questions/657 ... tensorflow