TensorFlow维度变换函数语句

学习课程
1.reshape

a = tf.random.normal([4,28,28,3]) #4张照片
a.shape,a.ndim  #TensorShape([4,28,28,3]),4
tf.reshape(a,[4,784,3]).shape  #TensorShape([4,784,3])
tf.reshape(a,[4,-1,3]).shape  #TensorShape([4,784,3])
tf.reshape(a,[4,-1]).shape  #TensorShape([4,2352])
tf.reshae(tf.reshape(a,[4,-1])[4,1,784,3]).shape  #TensorShape([4,1,784,3])

Reshape could lead to potential bugs!

2.tf.transpose

a = tf.random.normal([4,28,56,3])
tf.transpose(a).shape   #TensorShape([3,56,28,4])
tf.transpose(a,perm=[0,1,3,2]).shape   #TensorShape([4,28,3,56])

3.Expand dim

a = tf.random.normal([4,35,8])
tf.expand_dims(a,axis=0).shape  #TensorShape([1,4,35,8])
tf.expand_dims(a,axis=1).shape  #TensorShape([4,1,35,8])
tf.expand_dims(a,axis=-1).shape  #TensorShape([4,35,8,1])
tf.expand_dims(a,axis=-4).shape  #TensorShape([1,4,35,8])

4.Squeeze dim

a = tf.random.normal([1,1,4,35,1,8])
tf.squeeze(a,axis=0).shape  #TensorShape([1,4,35,1,8])
tf.squeeze(a,axis=4).shape  #TensorShape([1,1,4,35,8])
tf.squeeze(a,axis=-2).shape  #TensorShape([1,1,4,35,8])

5.Broadcasting

x=tf.random.normal([4,32,32,3])
x+tf.random.normal([3]).shape  #TensorShape([4,32,32,3])
x+tf.random.normal([32,32,1]).shape  #TensorShape([4,32,32,3])
x+tf.random.normal([4,1,1,1]).shape  #TensorShape([4,32,32,3])

6.tf.broadcast_to

b=tf.broadcast_to(tf.random.normal([4,1,1,1]),[4,32,32,3]).shape    #TensorShape([4,32,32,3])