如何解决为什么numba.prange函数不能与元组参数一起正常工作?
我正在尝试在Visual Studio代码上运行以下代码:
from numba import njit,prange
import numpy as np
@njit(parallel = True)
def parallel_sum(A):
sum = 0.0
for j in prange(A.shape[0]):
sum += A[j]
return sum
X = np.random.normal(size=(10,3))
F = np.zeros(X.shape[0])
for i in range(X.shape[0]):
F[i] = parallel_sum(X[i,:])
print(F[i])
我收到一条错误消息
不可迭代的值prange(A.shape [0])用于迭代上下文中
我知道A.shape [0]返回一个元组值。因此,我尝试了以下代码:
def parallel_sum(A):
sum = 0.0
k = int(A.shape[0])
for j in prange(k):
sum += A[j]
return sum
但是问题仍然存在。 stackoverflow.com where中有一个类似的问题,讨论了一种解决方案。但是,此解决方案也不起作用。我也没有在互联网上找到任何其他资源。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。