美文网首页
tensorflow的tf.reshape()的-1问题

tensorflow的tf.reshape()的-1问题

作者: theo_NI | 来源:发表于2017-12-11 23:19 被阅读0次
>>> a = tf.array([[1,2,3], [4,5,6]])
>>> tf.reshape(a, (3,-1))  # the unspecified value is inferred to be 2
array([[1, 2],
       [3, 4],
       [5, 6]])

-1相当于一个未知数,可以把它当做x,即tf.reshape(a,(3,x))
程序内部会自动计算x的值,并自动变形。

相关文章

网友评论

      本文标题:tensorflow的tf.reshape()的-1问题

      本文链接:https://www.haomeiwen.com/subject/fnawixtx.html