好的,我需要编写 pytorch LSTM 脚本,因为当我跟踪它时,它会通过不同的输入得到不同的答案。但是当我这样做时:
script_module=torch.jit.script(torch.nn.modules.rnn.LSTM(768, 256, num_layers=2, batch_first=True, bidirectional=True))
生成的 script_module 打印为:RecursiveScriptModule(original_name=LSTM),但与其他递归脚本模块不同(例如,此处生成:https ://pytorch.org/tutorials/beginner/Intro_to_TorchScript_tutorial.html ,并由我成功复制)它没有.code 或 .graph 属性,不能调用。
那么我做错了什么?它显然“认为”它产生了一个正确的脚本。