我有一个训练有素的张量流模型,现在我需要将其移植到Android平台上。但是我使用inferenceInterface.feed()函数有问题。在我发现的示例代码中,该函数只有一个输入节点。
private static final String INPUT_NODE = "input:0"; // input tensor name
private static final String OUTPUT_NODE = "output:0"; // output tensor name
private static final String[] OUTPUT_NODES = {"output:0"};
private static final int OUTPUT_SIZE = 10; // number of classes
private static final int INPUT_SIZE = 784; // size of the input
INPUT_IMAGE //MNIST Image
float[] result = new float[OUTPUT_SIZE]; // get the output probabilities for each class
inferenceInterface.feed(INPUT_NODE, INPUT_IMAGE, 1, INPUT_SIZE); //1-D input (1,INPUT_SIZE)
inferenceInterface.run(OUTPUT_NODES);
inferenceInterface.fetch(OUTPUT_NODE, result);
当我需要使用多个输入时,应该如何使用inferenceInterface.feed函数?我应该为每个输入调用一次此功能吗?如果尺寸不同,该怎么办?