tf.raw_ops.Squeeze

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

tf.raw_ops.Squeeze(
    input, axis=[], name=None
)

При данном тензоре input, эта операция возвращает тензор того же типа со всеми размерностями размера 1, удаленными. Если вы не хотите удалять все размерности размера 1, вы можете удалить определённые размерности размера 1, указав axis.

Например:

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
shape(squeeze(t)) ==> [2, 3]

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

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
shape(squeeze(t, [2, 4])) ==> [1, 2, 3, 1]