PyTorchでnn.LSTMCell
をONNXにエクスポートすると、LSTM operatorになるでもなく、LSTMCell operatorになるでもなく、プリミティブ opsの集まりとして出力される。
まず、ONNXにはRecurrentCellに相当するoperatorがない(参考: wikipedia/ONNX)。
なので nn.LSTM
はLSTMとしてexportされるのかな、が第一感。
しかしバラされる運命にある、理由はよくわからん。
PyTorchでnn.LSTMCell
をONNXにエクスポートすると、LSTM operatorになるでもなく、LSTMCell operatorになるでもなく、プリミティブ opsの集まりとして出力される。
まず、ONNXにはRecurrentCellに相当するoperatorがない(参考: wikipedia/ONNX)。
なので nn.LSTM
はLSTMとしてexportされるのかな、が第一感。
しかしバラされる運命にある、理由はよくわからん。