Different shape of input (1, C, H, W)

[longlivelonger]

Hi, my tflite model converted from Pytorch has input shape [1, C, H, W]. But TensorImage always gives me a tensor with shape [H, W, C]. I just gave up on reading source code. Please could you give me a clue on how to moveaxis or another solution to this case?

InputTensor of model has shape [1, 3, 240, 240] .
OutputTensor has shape [1,1].
Its regression task.

[SmiffyKMc]

Have you managed to progress on this?