numba 相关问题

numba 中的 numpy nanmean

我正在尝试为 numba 编写一个更简单的 numpy.nanmean 版本。 这是我的代码： 从 numba 导入 jit、prange 将 numpy 导入为 np @jit(nopython=True) def nanmeanMY(a, 轴=无): 如果 a.ndim&g...

提高批量马哈拉诺比斯距离计算的性能速度

我有以下一段代码，用于计算一组批处理特征上的马哈拉诺比斯距离，在我的设备上大约需要 100 毫秒，其中大部分是由于德之间的矩阵乘法...

有什么方法可以加快这个 numba 函数的速度，其相当于带有 o3 标志的 C++ 实现速度快 10 倍

函数 get_cum_dist 定义如下： 从 numba 导入 njit 将 numpy 导入为 np @njit(fastmath=True) def get_cum_dist(perm: np.ndarray, c: np.ndarray, n: int) -> np.array: 暨_距离...

使@intrinsic返回一个元组

我正在尝试在 numba 中实现 (uint64 a)*(uint64 b)->(uint64 upper,uint64 lower) 函数。这就是我整晚都在奋斗的事情。网上关于这方面的例子确实很少： 导入...

我无法使用 pip install 在 Python 中安装“numba”模块

不起作用“pip install numba”（python中的模块） 日志： C:\Users\Admin>pip 安装 numba 收集麻布 使用缓存的 numba-0.58.1.tar.gz (2.6 MB) 正在安装构建依赖项... ...

使用具有可变数量输入和输出参数的 numba 实现逐元素 ufunc

我正在用Python开发一个系统模拟器。我试图允许使用逐元素函数的自定义系统块。 Numpy 提供了一个函数“frompyfunc”来创建通用函数...

在 Python/Numba 中添加不同形状的矩阵的最快方法

我想“添加”两个矩阵，一个形状为 (K,T) 的矩阵 a 和一个形状为 (K,N) 的矩阵 b，以得到形状为 (K,T,N) 的矩阵 以下工作正常： 将 numpy 导入为 np 来自 numba imp...

如何将包含函数的数组作为参数传递给njit？

我想将一个包含 numba 编译函数列表的数组作为参数传递到 njit 方法中。在尝试这样做时，我遇到了以下错误：非精确类型数组（

在 Google 云平台上部署通过 Numba 编译的 Python 函数：必须维护状态

我正在使用 Nixtla 的 StatsForecast 库对多个产品运行时间序列预测。这是一座辉煌的图书馆。它最吸引人的特点是它的速度和可靠性

在 Azure Databricks 中导入 Whisper 会导致 Numba 出现导入错误，尽管安装了正确版本的 NumPy

我想在 Azure Databricks 中实现 Whisper 来执行转录任务。安装 Whisper 效果很好。使用 import Whisper 导入模块时，会出现以下 ImportError：Numba 需要

在 Numba 中使用 `numpy.random.normal()` 时出错

我正在探索一些 Numba 来优化一些信号处理代码。根据 Numba 的文档，即时编译器很好地支持 numpy.random 包中的函数。嗬...

使用多个条件优化 SmoothStep 的 Python 函数以进行 Numba 矢量化

我实现了一个使用 SmoothStep 创建平滑矩形函数的函数： 将 numpy 导入为 np 从 numba 导入 jit, njit 将 matplotlib.pyplot 导入为 plt @njit def GenSmoothStep( vX: np.

使用 numba 的代码比不使用 numba 的代码慢

我使用 Numba 的代码比不使用 Numba 的代码慢，我不知道发生了什么。请指导我。 #与numba 将 numpy 导入为 np 从 numba 导入 njit, prange 导入时间 （X_列车，Y_列车），（X...

在 numba cuda 中删除数组的零值

我有一个很长的数组 arr = np.array([1,1,2,2,3,3,0,0,2,2]) 我想在 numba cuda 中删除该数组的所有零值，因为实际数组非常大并且 numpy 非常慢。 有人吗

不使用 Numpy 进行矩阵求逆

我想在不使用 numpy.linalg.inv 的情况下反转矩阵。 原因是我使用 Numba 来加速代码，但不支持 numpy.linalg.inv，所以我想知道是否可以反转矩阵...

如何使用 jukeboxAI 错误 ERROR: Could not build Wheels for numba, llvmlite, 这是安装基于 pyproject.toml 的项目所必需的

我正在尝试通过 openAI 运行这个 colab 构建：https://github.com/openai/jukebox 你可以在github上看到colab链接：https://colab.research.google.com/github/openai/jukebox/blob/master/jukebox/

pandas：使用 numba 进行 groupby 应用

使用pandas v1.1.0。 在 pandas 文档中，有一个关于如何使用 numba 加速 moving.apply() 操作的很好的示例。 将 pandas 导入为 pd 将 numpy 导入为 np def mad(x): 返回 np.fa...

确定 jitclass 方法的输入参数类型

我正在开发一个 jitclass，其中方法之一可以接受 int、float 或 numpy.ndarray 的输入参数。我需要能够确定参数是数组还是其他两个中的任何一个

如何使用 nopython true 检查推断的 numba 类型？

没有 nopython=True numba.typeof 报告变量类型。 如何使用 nopython=True 检查变量的类型？ 导入numba @numba.jit(nopython=True) def f(): a = 0 打印（numba.typeof（a...

无法在另一个即时编译函数中调用存储的即时编译函数

Python 3.11.4 麻木 0.57.1 我正在构建一个“编译器”类，从现有代码生成 numba njit-ed 函数 导入numba 从输入 import Callable 类 Foo: def __init__(s...