pytorch lstmの内部状態を 取り出し 入れ替えたい

Question

やりたいこと
入れ替える　
pytorch1.5で
今の内部状態を別のところに保存し
すでに保存してある別の内部状態を突っ込みたい

rnnなら最後のoutputを取り出せばいいでしょうが
lstmの内部状態は 短期記憶　長期記憶　二つあるせいでできません
どうすればいいでしょうか？

試したこと
pytorchの関数などを検索しましたがわかりませんでした
lstmを実装してるところ見ればわかるかと思いましたがうまくいかず

なにか内部状態を取り出す　方法　関数などが分かる人教えてください

Answer 1

どうやら普通に内部状態を取り出す仕様らしいですね