一、环境
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_