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]