如题,模型的输出是一个三维矩阵,我需要通过TensorFlowInference接口中的fetch方法获得三维矩阵,再通过DoubleMatrix进行三维矩阵的分析,但是看fetch的源码,都是获得的各种一维数组,想请问各位大神有没有方法直接fetch到三维矩阵?
直接传不行,用reshape先把你的矩阵转换下,再传给TensorFlowInference接口