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

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张量。