Numba 是否支持“multi_dot”方面的多字典乘法?

问题描述

我需要使用 numpy 将三个大小为 25x25、25x60 和 60x60 的二维矩阵相乘以得到大小为 25x60 的结果。为了快速乘法,我想使用 hetemero(['great','dgss','may']) hetemero(['a25a','agd','ss25']) [('great',True),('dgss',False),('may',False)] [('a25a',('agd',('ss25',True)] 并尝试使用 from numpy.linalg import multi_dot 指令在 GPU 中并行执行它。

但是,在我的程序中,我收到以下 numba 错误

@jit(nopython=True)

当我不使用函数定义上方的 TypingError: Failed in nopython mode pipeline (step: nopython frontend) Untyped global name 'multi_dot': cannot determine Numba type of <class 'function'> 命令时,不会发生此错误,即按顺序它对我来说工作正常。

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)