forked from jiuyuan/InfiniTensor
YdrMaster
d9e2953425
- 从 initializer 拿到 reshape 的 shape 值 - 但 reshape 仍然无法导入，因为无法分辨 shape 其实不是一个后端张量 Signed-off-by: YdrMaster <ydrml@hotmail.com> |
||
|---|---|---|
| .. | ||
| docs | ||
| src/pyinfinitensor | ||
| tests | ||
| README.md | ||
| pyproject.toml | ||
README.md
pyinfinitensor
InfiniTensor 的 Python 前端。
- 从 ONNX 导入模型
- 从 Pytorch 导入模型
- 从 PaddlePaddle 导入模型
- 模型导出到 ONNX
- 模型导出到 Pytorch
- 模型导出到 PaddlePaddle
python 工程结构及打包方法
本项目使用 [pyproject.toml] 文件定义，目录结构采用 src 布局，后端库会被打包，并支持自动的依赖安装。
参考 setuptools 的文档。