Exporting LSTM to ONNX model at Operator level

1

Dear all,

I have a question regarding exporting the PyTorch model to ONNX at the operator level.

Regrading LSTM Cell, LSTM is layer level and matmul, add inside LSTM is operator level.

Currently, nn.LSTM is exported at the layer level, but is it possible to export operator level?

Any comments will be helpful.

Thanks,
Jaewook