einsum算子不支持问题处理

当torch.einsum算子在onnx转om找不到时,如何进行代码替换。
torch.einsum里面为
bhwc, hkc->bhwk
计算方式为bhwc乘以hkc得到bhwk

b h w c
  h k c
b h w k
   ||
   \/
b h w c
  h c k(交换c和k的位置,bhwc*ck=bhwk)
b h w k

举例:

M = torch.randn(100, 30, 40, 60)
N = torch.randn(30, 40, 60)
torch_transform_result = torch.einsum("bhwc,hkc->bhwk", M, N)

## 可以替换为torch.matmul
N_t = torch.transpose(N, 1, 2)
result = torch.matmul(M, N_t) ## 此结果和torch_transform_result是一样的。

参考:

https://blog.csdn.net/bj_zhb/article/details/136869289
https://pytorch.org/docs/stable/generated/torch.matmul.html#torch.matmul

相关推荐

  1. einsum算子支持问题处理

    2024-07-15 22:02:02       21 阅读
  2. 服务器支持PUT和DELETE请求处理

    2024-07-15 22:02:02       36 阅读
  3. 处理目标检测中的类别均衡问题

    2024-07-15 22:02:02       46 阅读
  4. torch.einsum用法

    2024-07-15 22:02:02       53 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-15 22:02:02       66 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-15 22:02:02       70 阅读
  3. 在Django里面运行非项目文件

    2024-07-15 22:02:02       57 阅读
  4. Python语言-面向对象

    2024-07-15 22:02:02       68 阅读

热门阅读

  1. Vant Ui 最新访问地址

    2024-07-15 22:02:02       18 阅读
  2. AWS云计算实战:电商平台发卡机器人开发指南

    2024-07-15 22:02:02       23 阅读
  3. lua 实现一个 StateMachine状态机

    2024-07-15 22:02:02       19 阅读
  4. vim全局替换、添加新行、同时注释多行

    2024-07-15 22:02:02       20 阅读
  5. 算法题——冒泡排序

    2024-07-15 22:02:02       16 阅读
  6. 构建加速秘籍:精通Gradle构建启动参数

    2024-07-15 22:02:02       20 阅读
  7. FATE Flow 源码解析 - 资源分配流程

    2024-07-15 22:02:02       17 阅读
  8. 每天一个数据分析题(四百二十九)- 假设检验

    2024-07-15 22:02:02       20 阅读
  9. Redis6.2.1版本集群新加副本

    2024-07-15 22:02:02       23 阅读
  10. Linux-线程同步(条件变量、POSIX信号量)

    2024-07-15 22:02:02       21 阅读