将输入传递给Android中的Tensorflow Lite模型

Question

我创建了一个神经网络，将数值数据作为输入，并使用python将其保存为tensorflow lite模型。 我正在尝试将输入传递给Android中的模型。 ndarray的形状为1 * 3

python中的输入示例如下

np.array([[-0.276786765 ,8.41897583008  ,-0.0222015380859]])

但是我不知道在java中创建相同的输入以将其传递给模型。

我尝试使用nd4j库。但是仍然无法编写适当的代码来创建模型所需的输入。

Answer 1

假设您正在使用TensorFlow Lite，则可以使用以下命令提供1x3输入：

float[] innerInput = {-0.276786765 ,8.41897583008  ,-0.0222015380859
float[][] input = {innerInput};
interpreter.run(input, output);