tensorflow中的reshape

a = tf.constant(np.arange(1, 13, dtype=np.int32),
                  shape=[2, 2, 3])
==>[[[ 1  2  3]
  [ 4  5  6]]

 [[ 7  8  9]
  [10 11 12]]]

b=tf.reshape(a,[-1,3])
==>[[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]]
阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页