将张量流矩阵的形状从(x,y,z)转换为(1,x,y,z)

问题描述:

如何将矩阵从3-D更改为4-D。更具体地说,如何将具有形状(x, y, z)的矩阵A转换为具有形状(1,x,y,z)的矩阵B将张量流矩阵的形状从(x,y,z)转换为(1,x,y,z)

,使得例如该元件A[7,3,5] == B[1,7,3,5]

+0

是的,我想从3的尺寸更改为4换言之,新的矩阵将是4d其中第一维是大小1 – BH85

tensorflow.expand_dims函数应该是完美匹配:

new_matrix = tf.expand_dims(old_matrix, axis=0)