Onnx错误：index: 0 Got: 120 Expected: 1

错误展示：

onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: input for the following indices
 index: 0 Got: 120 Expected: 1
 Please fix either the inputs or the model.

错误原因：

导出onnx模型的时候的输入数据和调用onnx模型进行推理时的输入数据不一致。

我这里导出onnx模型是的输入为：

x = torch.randn(1, 30, 42).to(device)

报错时的模型输入为：

x = torch.randn(120, 30, 42).to(device)

对比这两行再看报错提示中的：

index: 0 Got: 120 Expected: 1

在0维期望得到1，实际输入的则是120；所以产生从这个错误。

如要解决这个问题，只需将实际输入的0维改为1即可。