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

2017-12-11  本文已影响0人  theo_NI
>>> 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的值,并自动变形。

上一篇下一篇

猜你喜欢

热点阅读