numba 相关问题

Numba是一个开源的NumPy感知的Python优化编译器。

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

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

回答 1 投票 0

有什么方法可以加快这个 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: 暨_距离...

回答 1 投票 0

使@intrinsic返回一个元组

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

回答 1 投票 0

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

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

回答 1 投票 0

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

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

回答 1 投票 0

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

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

回答 1 投票 0

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

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

回答 1 投票 0

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

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

回答 1 投票 0

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

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

回答 1 投票 0

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

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

回答 1 投票 0

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

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

回答 1 投票 0

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

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

回答 1 投票 0

在 numba cuda 中删除数组的零值

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

回答 2 投票 0

不使用 Numpy 进行矩阵求逆

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

回答 8 投票 0

如何使用 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/

回答 1 投票 0

pandas:使用 numba 进行 groupby 应用

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

回答 3 投票 0

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

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

回答 5 投票 0

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

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

回答 2 投票 0

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

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

回答 1 投票 0

Numba Cuda 支持任何键值数据结构吗?

Numba 有一个 typed_dict 实现,但它似乎仅适用于 CPU 编译的代码。 Numba.cuda 是否支持任何键值数据结构(例如搜索树)?

回答 1 投票 0

© www.soinside.com 2019 - 2024. All rights reserved.