Java中的Tensorflow：inferenceInterface.fetch轉換為多維數組

Question

我在Android上的Java中使用訓練有素的Tensorflow模型。 我正在嘗試提取中間操作的輸出。

我提取的張量具有形狀（150、150、256）。

我已經聲明輸出目標為

private float[] hybridValues;
hybridValues = new float[150 * 150 * 256];

然后，我得到使用的輸出。

inferenceInterface.fetch(OUTPUT_NODE, hybridValues);

這些值很好，但是它們存儲為一維數組。 有沒有辦法讓inferenceinterface.fetch返回多維數組？

我嘗試將hybridValue聲明為三維浮點數組，但由於fetch方法期望使用1D數組，因此該方法不起作用。

最終目標是將我的輸出傳遞給Python程序，該程序會將值饋入相同形狀（150、150、256）的Tensor。

為了進行比較，Python a_output = graph.get_tensor_by_name('a2b_generator/Conv_7/Relu:0')返回一個ndarray，其值與目標張量的形狀相同。

Answer 1

我在桌面上使用tensorflow和java（可能有所不同），我要做的就是創建一個具有正確大小的多維數組，然后將值復制到上面，例如Tensor.copyTo(Object dst)