本篇文章将尝试了解 PyTorch 组件的高级概述以及它们如何配合。
PyTorch 组件的高级概述
PyTorch 后端是用 C++ 编写的,它提供 API 来访问高度优化的库,例如:用于高效矩阵运算的张量库、用于执行 GPU 运算的 CUDA 库以及用于梯度计算的自动微分等。
可以使用 Python 或 C++ 前端 ( libpytroch )访问 C++ 后端 API 接口来构建、训练和测试 ML/DL 模型。
京公网安备 11010502049817号