TensorFlow函数：tf.squeeze

tf.squeeze函数,

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

从张量形状中移除大小为1的维度。

给定一个张量 input，该操作返回一个与已经移除的所有大小为1的维度具有相同类型的张量。如果您不想删除所有大小为1的维度，则可以通过指定 axis 来删除特定的大小为1的维度。

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
tf.shape(tf.squeeze(t))  # [2, 3]

或者，要删除特定的大小为1的维度：

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
tf.shape(tf.squeeze(t, [2, 4]))  # [1, 2, 3, 1]

函数参数：
- input：A Tensor。该input挤。

• axis：一个可选列表ints。默认为[]。如果指定，只能挤压列出的尺寸。维度索引从0开始。压缩非1的维度是错误的。必须在范围内[-rank(input), rank(input))。
• name：操作的名称（可选）。
• squeeze_dims：现在是轴的已弃用的关键字参数。

函数返回值：
- Tensor。与。类型相同input。包含与之相同的数据input，但删除了一个或多个尺寸为1的尺寸。

可能引发的异常：
- ValueError：当两个squeeze_dims和axis指定。