tf.compat.v1.expand_dims

Возвращает тензор с осевой размерностью 1, вставленной по индексу axis. (устаревшие аргументы)

tf.compat.v1.expand_dims(
    input, axis=None, name=None, dim=None
)

Используется в блокнотах

Для данного тензора input, эта операция вставляет измерение длиной 1 в измерение с индексом axis в форме input. Индекс измерения следует правилам индексирования Python: он нулевой, а отрицательный индекс подсчитывается назад с конца.

Эта операция полезна для:

• Добавление внешнего измерения «пакета» к одному элементу.
• Выравнивание осей для трансляции.
• Добавление внутреннего измерения длины вектора к тензору скаляров.

Например:

Если у вас есть один образ с формой [height, width, channels]:

image = tf.zeros([10,10,3])

Вы можете добавить внешнее batch измерение, передав axis=0:

tf.expand_dims(image, axis=0).shape.as_list()
[1, 10, 10, 3]

Новое положение оси соответствует правилам Python list.insert(axis, 1):

tf.expand_dims(image, axis=1).shape.as_list()
[10, 1, 10, 3]

Согласно стандартным правилам индексирования Python, отрицательный axis отсчитывается от конца, поэтому axis=-1 добавляет внутреннее крайнее измерение:

tf.expand_dims(image, -1).shape.as_list()
[10, 10, 3, 1]

Эта операция требует, чтобы axis был допустимым индексом для input.shape, следуя правилам индексирования Python:

-1-tf.rank(input) <= axis <= tf.rank(input)

Эта операция связана с:

• tf.squeeze, который удаляет измерения размера 1.
• tf.reshape, который предоставляет более гибкие возможности изменения формы.
• tf.sparse.expand_dims, который предоставляет эту функциональность для tf.SparseTensor