tf.squeeze

参考  tf.squeeze - 云+社区 - 腾讯云

squeeze(
    input,
    axis=None,
    name=None,
    squeeze_dims=None
)

用于压缩张量中为1的轴，去除张量中形状为1的轴。

例：

import tensorflow as tf

raw = tf.Variable(tf.random_normal(shape=(1, 3, 2)))
squeezed = tf.squeeze(raw)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(raw.shape)
    print(sess.run(squeezed).shape)


Output:
-------------
(1, 3, 2)
(3, 2)
-------------

转载地址：tf.squeeze()用于压缩张量中为1的轴_机器学习杂货铺1号店-CSDN博客