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