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