快捷方式

ScriptFunction

class torch.jit.ScriptFunction

功能上等同于 ScriptModule,但它表示单个函数,且没有任何属性或参数。

get_debug_state(self: torch._C.ScriptFunction) torch._C.GraphExecutorState
save(self: torch._C.ScriptFunction, filename: str, _extra_files: dict[str, str] = {}) None
save_to_buffer(self: torch._C.ScriptFunction, _extra_files: dict[str, str] = {}) bytes

文档

查阅 PyTorch 全面的开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并获得问题解答

查看资源