numba 相关问题

当 numba jitclass 包含 jitted 函数时，如何指定它的字段？

我想创建一个 numba jitclass，其属性包含任何 jitted 函数。 # 在另一个文件中定义的简单 jitted 函数 @njit def my_function(x): x = x + 1 返回x ...

记录类型为字符串的numba jitclass

v3 变量是字符串值。我无法使用下面的代码运行，这会出现错误。 将 numpy 导入为 np 将 pandas 导入为 pd 从 numba.experimental 导入 jitclass 从 numba 导入类型 导入操作系统 ...

将 jit 与 numpy linspace 函数一起使用时出错

我正在尝试使用 jit 来编译我的代码。目前，除了 Numba 之外我唯一使用的库是 Numpy。当指定 nopython = True 时，它会抛出一堆 linspace 数组错误。我有

传递类变量是否会停止 Numba 中的并行化？

我有一个包装方法来调用 Numba 兼容函数。在下面的代码中，方法 get_neighbours_wrapper() 只是调用 Numba 函数 get_neighbours_Numba() 的包装器。 我想...

如何有效地加快 Pandas 中我心爱代码的循环速度？

我一直在考虑使用多处理、cyton 和 numba 来加速我的代码的以下部分，但我就是无法弄清楚。我目前的结论是我的代码是无效的，尽管它

使用 numba 进行 PI 的蒙特卡洛模拟对于最少的点数来说是最慢的？

作为家庭作业练习的一部分，我正在用 Python 实现 pi 的蒙特卡罗模拟。我正在使用 Numba 来加速和并行化计算。从我之前进行的测试中，我发现...

numba 按类型调度

我想在 numba 函数中分派第二个参数的类型，但失败了。 如果它是一个整数，那么应该返回一个向量， 如果它本身是一个整数数组，...

为什么这个带有列表输入的简单 numba 函数这么慢

进口numba 从输入导入列表 @numba.njit def test(a: List[int]) -> int: 返回1 测试（[i 为 i 在范围（2_000_000）]） 需要 2 秒并随着列表的大小线性缩放。 瓦拉皮...

如何并行化/加速令人尴尬的并行 numba 代码？

我有以下代码： @nb.njit(缓存=True) def find_two_largest(arr): # 初始化第一大和第二大元素 如果 arr[0] >= arr[1]： 最大= arr[0]

将数组参数发送到 numba 函数

我有以下代码，它将数组作为参数发送到 numba 函数： 将 numpy 导入为 np 从 numba 导入 njit，float64 A = [(0.0182286178413157,-1.2904019395416308), ( 0.

如何使用numba加速多维Longsumexp和softmax

我想使用多项式logit计算softmax/概率，同时使用longsumexp以避免溢出。使用 numba 带来 2-3 倍的加速。我在这里可以做得更好吗？另外，当我使用 fastmath=True 时，我...

将 2d 字符串 numpy 数组转换为 3d int 数组的快速方法

我有一个非常大的 numpy 数组，其中包含以下条目： [['0/1''2/0'] ['3/0''1/4']] 我想转换它/获取一个带有 3d 数组的数组，例如 [[[0 1] [2 0]] [[3 0] [1 4]]] 数组很宽，所以...

如何使用Numba CUDA JIT装饰器？

我已按照本教程使用 Numba CUDA JIT 装饰器：https://www.youtube.com/watch?v=-lcWV4wkHsk&t=510s。 这是我的Python代码： 将 numpy 导入为 np from timeit import default_timer as t...

jit - 尽管没有在 numba 中使用 nopython，但“在 nopython 模式管道中失败”错误

用于 SciPy 集成和插值的 Numba

我使用 Numba 来加速我的代码。它效果很好，可提供 2-3 倍的改进。然而，我的代码中花费的主要时间（大约 90%）是在 scipy 四边形积分和插值上......

如何使用 numba 使用数组中的值重新定义积分来加速积分

我尝试使用 python numba 更快地计算积分。尽管使用 numba 的计时速度对于单次计算来说几乎快了 10 倍，但当我循环 redefini 的过程时......

最快的双积分方法

我正在使用 scipy 的双积分 dblquad，我正在尝试提高速度。我已经检查了网上提出的解决方案，但无法使它们发挥作用。 为了缓解这个问题，我准备了......

在 Python 函数中，如何判断它是在 GPU 中使用 Numba 执行还是在主机/CPU 上定期调用？

我有一个函数，有时会使用 Numba 作为在 GPU 上执行的设备函数进行调用，有时我会直接从主机上的常规 Python 中调用： 定义过程（）： # 执行计算...

将 numba jitclass 中的 numpy datetime64 转换为 unix 时间戳

为了可读性，我希望能够向 numba jitclass 提供 numpy.datetime64 对象，该对象在类本身内转换为浮点格式的 unix 纪元时间戳。 我目前

如何在 Numba 的 np.repeat() 中重新创建 axis = 2

我正在尝试将我的代码转换为 Numba 友好的实现，但是我不断遇到 axis 参数的错误（因为它不受支持）。具体来说，我需要使用 np.repeat() 函数...