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 — количество элементов в сжатых размерностях.