python - 使用 @jitclass 装饰器进行类实例化
问题描述
我正在尝试使用 Numba 的 @jitclass 装饰器,显然是为了加快我的代码速度。我收到的错误似乎是为了对@jitclass 有基本的了解
ss 装饰师。
import numba
from numba import jitclass
spec = [('raster',numba.float32[:,:]),('height', numba.int32),
('width', numba.int32),('azis', numba. int64[:]),('grid',numba.int32),('rough',numba.float32[:,:])]#,('maxrange',numba.float32[:,:]),('azir atio',numba.float32[:,:]),('labels',numba.int32[:,:])]
@jitclass(spec)
class raster_class(object):
def __init__(self,raster):
self.raster = raster
self.height =self.raster.shape[0]
self.width = self.raster.shape[1]
self.azis = np.arange(0,170,10)
self.grid = 500
x,y = numba.int32(self.height/self.grid),numba.int32(self.width/self.grid)
self.rough = np.zeros((x,y),dtype=np.float32)
self.maxrange = np.zeros((self.height/self.grid),(self.width/self.grid))
self.aziratio = self.rough.copy()
self.labels = self.rough.copy()
if __name__ == '__main__':
raster = np.zeros(1000,1000)
raster_class = raster_class(raster)
产生的错误:
TypingError: Failed in nopython mode pipeline (step: nopython frontend)
Internal error at <numba.typeinfer.CallConstraint object at 0x7f19b3ce5128>:
File "processraster.py", line 24:
def __init__(self,raster):
<source elided>
self.azis = np.arange(0,170,10)
self.grid = 500
^
[1] During: lowering "$0.45 = call $0.41($0.44, func=$0.41, args=[Var($0.44, /home/dunbar/DEM/processraster.py (24))], kws=(), vararg=None)" at /home/dunbar/DEM/processraster.py (24)
[2] During: resolving callee type: jitclass.raster_class#7f19a049f828<raster:array(float32, 2d, A),height:int32,width:int32,azis:array(int64, 1d, A),grid:int32,rough:array(float32, 2d, A)>
[3] During: typing of call at <string> (3)
--%<----------------------------------------------------------------------------
File "<string>", line 3:
<source missing, REPL/exec in use?>
解决方案
我用numba
的不多,这个jitclass
也没有,但是通过阅读文档和一些试验和错误,我得到了这个工作:
In [308]: spec = [('raster',numba.float32[:,:]),('height', numba.int32),('width', numba.int32),('azis', numba.int64[:])]
In [309]: @numba.jitclass(spec)
...: class raster_class():
...: def __init__ (self,raster):
...: self.raster = raster
...: self.height = self.raster.shape[0]
...: self.width = self.raster.shape[1]
...: self.azis = np.arange(0,170,10)
...:
请注意,spec
包括所有字段/属性,而不是参数。
In [310]: raster_class(np.zeros((3,2),'float32'))
Out[310]: <numba.jitclass.boxing.raster_class at 0x7fe67f11f090>
In [311]: foo = _
In [312]: foo.width
Out[312]: 2
In [313]: foo.azis
Out[313]:
array([ 0, 10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120, 130, 140, 150, 160])
对 dtypes也numba
很挑剔。
推荐阅读
- angular - NG 测试错误“找不到模块...”
- c++ - 与松弛原子同步
- rest - Firebase REST API 使用电话号码进行身份验证
- r - 使用库(odbc)更新 R 中的 sql 表返回错误:外部指针无效
- neo4j - 查找不在特定模式中的节点数 [Neo4j]
- apache - 如何在apache中将任何网页重定向到另一个网页
- asp.net-core - 开发时在 ASP.NET Core 2.1 中使用 SSL 证书
- hbase - 使用 HBase 扫描的 ScanMetrics 中的 countOfRowsFiltered 到底是什么?
- python - 使用 OpenCV 合并重叠矩形
- vba - Excel VBA循环通过选项卡复制并粘贴到另一个文件中的一个单独的文件中......