TensorFlow函数教程：tf.keras.backend.squeeze

tf.keras.backend.squeeze函数

tf.keras.backend.squeeze(
    x,
    axis
)

定义在：tensorflow/python/keras/backend.py。

从索引“axis”处的张量中移除1维。

参数：

• x：张量或变量。
• axis：要下降的轴。

返回：

与x的数据相同但维数减少的张量。