PyTorch 是一种流行的机器学习开源框架，已更新至版本 1.2.0。 新版本包含 1900 多个修复，涵盖 JIT、ONNX、分布式学习模式和性能改进。
一些变化：
- 新的 TorchScript API 它允许 将 nn.Module（包括在forward() 中调用的子模块和方法）转换为 ScriptModule 很容易。
- 与 Microsoft 一起添加了对 ONNX Opset 版本 7 (v1.2)、8 (v1.3)、9 (v1.4) 和 10 (v1.5) 的全面支持。 此外，用户现在可以注册自己的符号以进行自定义操作导出，并在导出期间指定动态输入大小。
- 张量板支持 不再是 实验性的。
- 在文章的基础上添加了 nn.Transformer 模块 注意就是您所需要的.
- 对 C++ API 进行了大量改进。
来源： linux.org.ru