java如何在安卓上设置tflite模型输入张量和输出张量？

Question

我有一个模型：

input shape: (batch_size, 3, 48, 64, 18)
output shape: (batch_size, 512)

我想将这个模型转换为tflite模型，并在安卓上运行它

tflite的官方示例显示，输入张量是一个one-dimensional ByteBuffer，输出张量是一个浮点数组

如何将形状（批次大小，3,48,64,18）的输入数据存储到ByteBuffer

在我的项目中，我设置了batch_size = 3，因为原始输入数据的形状是（54,48,64,18），我使用numpy.reshape将数据重新塑造为（3,3,48,64,18）

它真的需要先做整形吗

我可以直接将带有形状（54、48、64、18）的数据存储到ByteBuffer吗