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]]
阅读更多
个人分类: Tensorflow Python
想对作者说点什么? 我来说一句

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

不良信息举报

tensorflow中的reshape

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭