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