tf.squeeze

2018-07-16  本文已影响8人  yalesaleng

参考:

tf.squeeze的作用是去掉维度为1的维,当然也可以去掉指定维度的值。

例如,

当前维度是(2,3,1,4)=>(2,3,4)
当前维度是(1,2,1,4)=>(2,4)
上一篇 下一篇

猜你喜欢

热点阅读