tf.squeeze

Удаляет измерения размера 1 из формы тензора. (устаревшие аргументы)

tf.squeeze(
    input, axis=None, name=None, squeeze_dims=None
)

Для данного тензора input, эта операция возвращает тензор того же типа со всеми измерениями размера 1, удалёнными. Если вы не хотите удалять все измерения размера 1, вы можете удалить определённые измерения размера 1, указав axis.

Например:

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
tf.shape(tf.squeeze(t))  # [2, 3]

Или, чтобы удалить определённые измерения размера 1:

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
tf.shape(tf.squeeze(t, [2, 4]))  # [1, 2, 3, 1]

Примечание: если input является tf.RaggedTensor, то эта операция занимает O(N) времени, где N — количество элементов в сжатых измерениях.