tf.squeeze()函数

tf.squeeze()函数是TensorFlow中的一个操作函数，它可以从张量中删除所有大小为1的维度，即将张量中所有维度大小为1的维度删除。例如，如果一个张量的形状为[1, 2, 1, 3, 1]，则使用tf.squeeze()函数处理后，它的形状将变为[2, 3]。该函数的语法如下：

tf.squeeze(input, axis=None, name=None)

其中，input：需要被处理的张量；axis：一个可选的整数列表，指定需要被删除的维度；name：可选的操作名称。如果不指定axis参数，则默认删除所有大小为1的维度。如果指定了axis参数，则只删除指定的维度，如果指定的维度不为1，则不会进行任何操作。