squeeze(
input,
axis=None,
name=None,
squeeze_dims=None
)
# 't' 是一个维度是[1, 2, 1, 3, 1, 1]的张量
tf.shape(tf.squeeze(t)) # [2, 3], 默认删除所有为1的维度
# 't' 是一个维度[1, 2, 1, 3, 1, 1]的张量
tf.shape(tf.squeeze(t, [2, 4])) # [1, 2, 3, 1],标号从零开始,只删掉了2和4维的1
该函数返回一个张量,这个张量是将原始input中所有维度为1的那些维都删掉的结果。