tf squeeze

tf.squeeze是TensorFlow中的一个函数，可以用来删除张量中尺寸为1的维度。默认情况下，tf.squeeze会删除所有尺寸为1的维度。如果不想删除所有尺寸为1的维度，可以通过指定axis参数来删除特定维度为1的维度。

参考代码如下：

import tensorflow as tf
import numpy as np

value = np.floor(10*np.random.random((1,3,2,1,2)))

with tf.Session() as sess:
    result = tf.squeeze(value) # 默认删除所有维度为1的维度
    print(sess.run(tf.shape(result))) # 输出结果 [3 2 2]

    result = tf.squeeze(value, [0]) # 删除第0个维度为1的维度
    print(sess.run(tf.shape(result))) # 输出结果 [3 2 1 2]