В тензорном режиме, какие оси 0, 1 и -1 переданы в функции типа argmax ()

Question

def accuracy(y_pred, y_true):
# Predicted class is the index of highest score in prediction vector (i.e. argmax).
correct_prediction = tf.equal(tf.argmax(y_pred, 1), tf.cast(y_true, tf.int64))
return tf.reduce_mean(tf.cast(correct_prediction, tf.float32), axis=-1)

Я начал изучать тензорный поток и с трудом понимаю, что такое ранг и ось тензорного потока. Может кто-нибудь объяснить, почему o, argmax() '1' передается, почему в redumean () -1 передается