关于在tensorflow中训练模型然后导入sdsoc软件里

在tensorflow中训练好了模型，现在想把模型导入sdsoc中，但是权值是一个4维矩阵，我现在想把它压缩成一维矩阵，然后在sdsoc中恢复成原来的矩阵，总是失败，希望有大佬来帮帮我，现在已经完成的是压缩成一维矩阵，但是不知道数据的排列顺序。

for(int batch_cnt=0; batch_cnt<image_Batch; batch_cnt++) {
        ROW_K:
        for(int row_k=0;row_k<5;row_k++){
            COL_K:
            for(int col_k=0;col_k<5;col_k++){
                ROW :
                for (int row = 0; row < CONV_1_OUTPUT_WH; row++) {
                    COL :
                    for (int col = 0; col < CONV_1_OUTPUT_WH; col++) {
                     #pragma HLS PIPELINE II=1
                        float input_pixel = IBRAM[batch_cnt][row+row_k][col+col_k];
                        float mult[8];
                        #pragma HLS array_partition variable=mult complete dim=0
                        D_OUT:
                        for(int co=0;co<8;co++){
                        #pragma HLS unroll
                            mult[co] = input_pixel*WBRAM[row_k][col_k][co];
                            if(row_k==0&&col_k==0)
                                OBRAM[batch_cnt][co][row*CONV_1_OUTPUT_WH+col] = mult[co];
                            else
                                OBRAM[batch_cnt][co][row*CONV_1_OUTPUT_WH+col] += mult[co];
                        }
                    }
                }
            }
        }
    }

https://blog.csdn.net/zichen_ziqi/article/details/80323560