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

在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];
                        }
                    }
                }
            }
        }
    }

1个回答

Csdn user default icon
上传中...
上传图片
插入图片
抄袭、复制答案,以达到刷声望分或其他目的的行为,在CSDN问答是严格禁止的,一经发现立刻封号。是时候展现真正的技术了!
立即提问