没有合适的资源?快使用搜索试试~ 我知道了~
opt_einsum_fx:使用opt_einsum和PyTorch FX图形重写进行Einsum优化
共12个文件
py:8个
license:1个
toml:1个
需积分: 37 0 下载量 169 浏览量
2021-03-16
00:26:12
上传
评论
收藏 15KB ZIP 举报
温馨提示
opt_einsum_fx 正在开发中 使用opt_einsum和PyTorch FX图形重写进行Einsum优化。 该库当前支持: 将多个枚举融合为一个 使用库优化einsum 用标量常量融合乘法和除法,包括通过与einsum之类的运算进行融合,这些操作与标量乘法相通。 安装 $ pip install 您可以使用以下命令运行测试 $ pytest tests/ 用法 opt_einsum_fx基于 ,该框架是用于在PyTorch Python代码和以编程方式可操纵的计算图之间进行转换的框架。 要使用此程序包,必须有可能将您的函数或模型作为torch.fx.Graph :讨论了FX符号跟踪的局限性。 最小的例子 import torch import torch . fx import opt_einsum_fx def einmatvecmul ( a , b , vec )
资源推荐
资源详情
资源评论
收起资源包目录
opt_einsum_fx-main.zip (12个子文件)
opt_einsum_fx-main
pyproject.toml 103B
opt_einsum_fx
__init__.py 161B
_fuse.py 12KB
_opt_ein.py 6KB
_script.py 1KB
tests
test_fuse.py 4KB
test_einsum_optimizer.py 3KB
conftest.py 932B
LICENSE 1KB
setup.py 872B
.gitignore 2KB
README.md 3KB
共 12 条
- 1
资源评论
还是那个小宇
- 粉丝: 25
- 资源: 4729
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功