tf.compat.v2.squeeze

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

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