tf.squeeze示例代码

官网上对此方法有介绍，但缺少验证示例，而百度上搜到的都是从官网上翻译过来的，正确倒都是正确的，只是缺少验证示例，这边写一个示例代码，希望能帮助理解

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.]]
 

大致意思就是去掉了很多方括号，(*^▽^*)