TASO

TASO: 适用于 深度学习 的张量代数超级优化器

请参阅 说明文档 以从源代码安装 TASO。 我们还提供了预先构建的 Docker 镜像，其中包含了所有依赖项的预先安装。

TASO 可以直接优化预训练的 ONNX 模型,这可以在几行 Python 代码中完成。下面的代码段展示了如何从 ONNX 加载预训练的 DNN 模型,优化模型,并将优化后的模型保存为 ONNX 文件。