我想用ONNX导出函数导出我的深度神经网络,并在Windows ML中使用它
WinMLRunner
,它只是检查.onnx文件。这似乎适用于cnn网络,但是对于lstm网络,我收到的错误是:“第一次输入没有排名2”。如果我在网络中有不止一个lstm层,错误消息就会以某种方式变成:“第一个输入张量必须有秩3”。
我附加了一个带有以下简单LSTM-network的文件和另一个带有相应ONNX导出的文件(Windows ML需要onset v7或v8):
层= [...
sequenceInputLayer (14)
lstmLayer (100“OutputMode”,“最后一次”);
fullyConnectedLayer (1)
regressionLayer];
我导出的网络如下:
exportONNXNetwork (lstmNet“lstm_net_onset_v7.onnx”,“OpsetVersion”7)
然后用WinMLRunner测试它:
WinMLRunner.exe模型C: \ lstm_net_onset_v7。onnx cpu
感谢您对如何解决这个问题提出的任何建议。我已经在ONNX转换器的文件交换页面上发布了这个问题。