title: How to set the tflite model input tensor and output tensor on android?
body: <p>I have a model with:</p>
  
  <pre><code>input shape: (batch_size, 3, 48, 64, 18)
  output shape: (batch_size, 512)
  </code></pre>
  
  <p>I want to convert this model to tflite model, and run it on android.</p>
  
  <p>The official example of tflite shows that the input tensor is an <code>one-dimensional ByteBuffer</code>, and the output tensor is a float array.</p>
  
  <p>How to store the input data with shape(batch_size, 3, 48, 64, 18) to the ByteBuffer?</p>
  
  <p>In my project, I set <code>batch_size = 3</code> because the shape of original input data is (54, 48, 64, 18), and I use <code>numpy.reshape</code> to reshape the data into (3, 3, 48, 64, 18).</p>
  
  <p>Does it really need to do the reshape first?</p>
  
  <p>Can I store the data with shape (54, 48, 64, 18) to the ByteBuffer directly?</p>