tensorflow去掉某一维度_TensorFlow 移除所有尺度为1的维度 tf.squeeze 的基本用法及实例代码...

一、环境

TensorFlow API r1.12

CUDA 9.2 V9.2.148

cudnn64_7.dll

Python 3.6.3

Windows 10

二、官方说明

从张量的形状中移除所有尺寸为1的维数。(弃用参数)

https://tensorflow.google.cn/api_docs/python/tf/squeeze

tf.squeeze(

input,

axis=None,

name=None,

squeeze_dims=None

)

参数：

input：要缩减维度的张量

axis：可选整型列表，默认为 [ ]，如果指定了给参数，值域列表中指定的维度会被移除。维度所以从 0 开始，范围是 [- rank(input), rank(input)]。不能移除尺度不为 1 的维度，否则会报错！

name：可选参数，设置操作的名称

squeeze_dims：被移除的关键字参数，通过 axis 替代

返回：

包含输入 input 中的数据，但移除了所有尺度为 1 的维度的张量，和输入 input 的数据类型相同

三、实例

(1)尺度缩减的错误方式

>>> raw_tensor = tf.constant(value=[[[1,2,3],[4,5,6]]])

>>> raw_tensor

>>> squeezed_