tf.squeeze

Удаляет измерения размера 1 из формы тензора.

tf.squeeze(
    input, axis=None, name=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]

В отличие от более старой операции tf.compat.v1.squeeze, эта операция не принимает устаревший squeeze_dims аргумент.

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