Определение оси тензорного потока начинается с 0 или 1?

Question

Я совершенно сбит с толку функцией уменьшения оси в Tensorflow. Например, reduce_max(). Предположим, у меня есть четырехмерная переменная a с формой a = [x, y, z, w], где x, y, z и w - длина этого измерения. У меня есть команда типа b = tf.reduce_max(a, axis=(1, 3)), в этом случае, какова будет форма b? b = [x, z] или b=[y, w]? По сути, мне просто интересно, на какой оси работает reduce_max(). Если ось определена от 0, то b = [x, z]. В противном случае b = [y, w].

Answer 1

Итак, у вас есть тензор a, с формой [x, y, z, w]. При использовании функции, которая ожидает параметр оси, нумерация оси для a будет [0, 1, 2, 3]. Уменьшение осей 1 и 3 приводит к тому, что эти оси «исчезают» (поскольку вы уменьшаете эти оси), поэтому выходные данные будут состоять из исходных осей [0, 2], что означает, что результирующая фигура будет [x, z].