Я использую обученную модель Tensorflow в Java на Android. Я пытаюсь извлечь вывод промежуточной операции.
Тензор, который я извлекаю, имеет форму (150, 150, 256).
Я объявил, что цель вывода будет
private float[] hybridValues;
hybridValues = new float[150 * 150 * 256];
Затем я получаю вывод с помощью.
inferenceInterface.fetch(OUTPUT_NODE, hybridValues);
Значения в порядке, но они хранятся в виде одномерного массива. Есть ли способ заставить inferenceinterface.fetch возвращать многомерный массив?
Я попытался объявить hybridValue трехмерным массивом с плавающей запятой, но это не работает, поскольку метод выборки ожидает одномерный массив.
Конечная цель состоит в том, чтобы передать мои выходные данные программе Python, которая передаст значения тензору той же формы (150, 150, 256).
Для сравнения Python a_output = graph.get_tensor_by_name('a2b_generator/Conv_7/Relu:0')
возвращает ndarray со значениями той же формы, что и целевой тензор.