たれぱんのびぼーろく

わたしの備忘録、生物学とプログラミングが多いかも

PyTorchでLSTMCellをONNXにエクスポートする: こいつ、ばらけるぞ

PyTorchでnn.LSTMCellをONNXにエクスポートすると、LSTM operatorになるでもなく、LSTMCell operatorになるでもなく、プリミティブ opsの集まりとして出力される。

gist.github.com

まず、ONNXにはRecurrentCellに相当するoperatorがない(参考: wikipedia/ONNX)。
なので nn.LSTM はLSTMとしてexportされるのかな、が第一感。
しかしバラされる運命にある、理由はよくわからん。