tensorflow - 将通道最后转换为通道第一图像

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