官网上对此方法有介绍,但缺少验证示例,而百度上搜到的都是从官网上翻译过来的,正确倒都是正确的,只是缺少验证示例,这边写一个示例代码,希望能帮助理解
import tensorflow as tf
k = tf.constant([[[[[[1]],[[2]],[[3]]]],[[[[4]],[[5]],[[6]]]]]])
with tf.Session() as sess:
print(sess.run(tf.shape(k)))
print("-----------------------")
print(sess.run(tf.squeeze(k)))
运行结果为
[1 2 1 3 1 1]
-----------------------
[[1. 2. 3.]
[4. 5. 6.]]
大致意思就是去掉了很多方括号,(*^▽^*)