學(xué)習(xí)課程

1.concat

a=tf.random_normal([4,35,8]) b=tf.random_normal([2,35,8]) c=tf.concak([a,b],axis=0) c.shape #TensorShape([6,35,8])a=tf.random_normal([4,30,8]) b=tf.random_normal([4,35,8]) c=tf.concak([a,b],axis=1) c.shape #TensorShape([4,65,8])

2.stack: create new dim

a=tf.random_normal([4,35,8]) b=tf.random_normal([4,35,8]) tf.concat([a,b],axis=-1).shape #TensorShape([4,35,16])tf.stack([a,b],axis=0).shape #TensorShape([2,4,35,8]) tf.stack([a,b],axis=3).shape #TensorShape([4,35,8,2])

3.Unstack

a=tf.random_normal([4,35,8]) b=tf.random_normal([4,35,8]) c=tf.stack([a,b]) c.shape #TensorShape([2,4,35,8]) aa,bb=tf.unstack(c,axis=0) aa.shape,bb.shape #TensorShape([4,35,8]),TensorShape([4,35,8])res=tf.unstack(c,axis=3) res[0].shape,res[7].shape #TensorShape([2,4,35]),TensorShape([2,4,35])

4.Split

a=tf.random_normal([4,35,8]) b=tf.random_normal([4,35,8]) c=tf.stack([a,b]) c.shape #TensorShape([2,4,35,8]) res=tf.split(c,axis=3,num_or_size_splits=2) len(res) ##2 res.shape[0] #TensorShape([2,4,35,4])res=tf.split(c,axis=3,num_or_size_split=[2,2,4]) res.shape[0],res.shape[1] ,res.shape[2] #TensorShape([2,4,35,2]),TensorShape([2,4,35,2]),TensorShape([2,4,35,4])

總結(jié)

以上是生活随笔為你收集整理的TensorFlow 合并与分割的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。