tensorflow的concat(),stack(),split(),unstack()合并及分割函数

tensorflow的concat()及stack()合并函数

1.concat()合并数据函数
合并两个相同维度的某个维度数据，其他维度个数相同看例子：

#contact在某一个维度上进行合并，合并后不会改变维度数
a=tf.ones([4,35,3])
b=tf.ones([2,35,3])
c=tf.concat([a,b],axis=0)
print(c.shape)#[6,35,3]

2. stack()合并数据函数
合并两个dim及shape均相同的数据，合并后会在指定位置增加维度

#stack 合并两个同样维度的tensor，合并之后会增加维度n=>n+1
a1=tf.ones([4,35,3])
b1=tf.ones([4,35,3])
c1=tf.stack([a1,b1],axis=0)
print(c1.shape)#[2,4,35,3]

3.unstack()分割数据函数
与stack()函数是可逆的过程函数

#unstack
a22,b22=tf.unstack(c1,axis=0)
print(a22.shape,b22.shape)#[4,35,3],[4,35,3]

4.split()分割数据函数

#split 可以自定义打散数据的shape
res=tf.split(c1,axis=3,num_or_size_splits=2)
print(len(res))#2
print(res[0].shape)#(2,4,35,4)

res=tf.split(c1,axis=3,num_or_size_splits=[2,2,4])
print(res[0].shape,res[2].shape)#(2,4,35,2)(2,4,35,4)