将频道的最后一个转换为频道的第一个图像

Question

我具有以下尺寸的图像张量

TensorShape([Dimension(1), Dimension(96), Dimension(96), Dimension(3)])

我希望该张量在通道的第一维中

TensorShape([Dimension(1), Dimension(2), Dimension(96), Dimension(96)])

我尝试过

tf.transpose (image, perm = [0,3,1,2])

但是它没有用。它和以前一样返回。

因为这是Facenet算法的要求，请提出实现方法。

Answer 1

您可以尝试将张量转换为NumPy数组，然后使用np.rollaxis并转换回Tensorflow张量。