たれぱんのびぼーろく

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

2022-06-07から1日間の記事一覧

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

PyTorchでnn.LSTMCellをONNXにエクスポートすると、LSTM operatorになるでもなく、LSTMCell operatorになるでもなく、プリミティブ opsの集まりとして出力される。 gist.github.com まず、ONNXにはRecurrentCellに相当するoperatorがない(参考: wikipedia/O…