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]