当前位置: 首页 > 文档资料 > PyTorch 中文文档 >

torch.legacy

优质
小牛编辑
138浏览
2023-12-01

此包中包含从Lua Torch移植来的代码。

为了可以使用现有的模型并且方便当前Lua Torch使用者过渡,我们创建了这个包。 可以在torch.legacy.nn中找到nn代码,并在torch.legacy.optim中找到optim代码。 API应该完全匹配Lua Torch。