tf.squeeze(
input,
axis=None,
name=None,
squeeze_dims=None
)
参数 | 描述 |
---|---|
input | (tensor) |
axis | (list)移除指定的维度,维度的开始索引为0 |
name |
将所有大小为1的维度移除,可以通过axis指定想要移除的维度.
tf.squeeze(
input,
axis=None,
name=None,
squeeze_dims=None
)
参数 | 描述 |
---|---|
input | (tensor) |
axis | (list)移除指定的维度,维度的开始索引为0 |
name |
将所有大小为1的维度移除,可以通过axis指定想要移除的维度.