I want to transpose the output shape of a graph in saved_model or tflite format. The current form looks like this.(.tflite)
I want to change the output of "partitionedcall:1" to [1,32,160,160], in this case do I need to add a layer of transpose after the current output layer? If you don't mind, can you tell me how to do it?
Thank you.