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)
-------------