阅读(6.9k) 书签 (0)

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

2019-04-14 20:32 更新

tf.keras.backend.squeeze函数

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

定义在:tensorflow/python/keras/backend.py

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

参数:

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

返回:

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