tf.squeeze
2018-07-16 本文已影响8人
yalesaleng
参考:
tf.squeeze的作用是去掉维度为1的维,当然也可以去掉指定维度的值。
例如,
当前维度是(2,3,1,4)=>(2,3,4)
当前维度是(1,2,1,4)=>(2,4)
参考:
tf.squeeze的作用是去掉维度为1的维,当然也可以去掉指定维度的值。
例如,
当前维度是(2,3,1,4)=>(2,3,4)
当前维度是(1,2,1,4)=>(2,4)