tf.raw_ops.ExpandDims

Добавляет измерение размера 1 в форму тензора.

tf.raw_ops.ExpandDims(
    input, axis, name=None
)

Для тензора input эта операция добавляет измерение размера 1 по индексу измерения axis в форму input. Отсчет индексов измерений axis начинается с нуля; если для axis указано отрицательное число, отсчет ведется с конца.

Эта операция полезна, если нужно добавить измерение пакета к отдельному элементу. Например, если у вас есть одно изображение формы [height, width, channels], с помощью expand_dims(image, 0) его можно преобразовать в пакет из одного изображения, форма которого будет [1, height, width, channels].

Другие примеры:

# 't' is a tensor of shape [2]
shape(expand_dims(t, 0)) ==> [1, 2]
shape(expand_dims(t, 1)) ==> [2, 1]
shape(expand_dims(t, -1)) ==> [2, 1]

# 't2' is a tensor of shape [2, 3, 5]
shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5]
shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5]
shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]

Для этой операции необходимо, чтобы:

-1-input.dims() <= dim <= input.dims()

Эта операция связана с squeeze(), которая удаляет измерения размера 1.