onnx 模型轉換 rknn 維度計算錯誤問題

大家好

我有訓練一個 pytorch 模型並且想經由RK1808運算棒做計算
目前做法是先將 pytorch 模型轉成 onnx 模型 (opset11) ，再將 onnx 模型轉至 rknn
onnx 模型經由測試資料驗證，轉換與模型輸出結果皆正常而將onnx 轉成 rknn 時，輸出維度錯誤，不符合原本輸出

根據轉換後的結果(如圖所示)
permute輸出(500x150x1x1) 結果符合原始模型，並與 variable (220x500) 做矩陣相乘(MATRIXMUL)
正常結果應為 220 x 150 x 1 x 1
但是轉換後卻輸出 500 x 500 x 1 x 1


請問這部分該如何處理 ?

rknn toolkit 跟 npu driver 皆為 1.7.0

謝謝