tf.compat.v1.squeeze

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

tf.compat.v1.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]
t = tf.ones([1, 2, 1, 3, 1, 1])
print(tf.shape(tf.squeeze(t)).numpy())
[2 3]

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

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

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