Python get_numba_ufunc示例

示例#1

文件： blaze.py 项目： kastnerkyle/dask

 def compute_broadcast(expr, *data, **kwargs):
     leaves = expr._inputs
     expr_inds = tuple(range(ndim(expr)))[::-1]
     func = get_numba_ufunc(expr)
     return atop(func,
                 next(names), expr_inds,
                 *concat((dat, tuple(range(ndim(dat))[::-1])) for dat in data))

示例#2

文件： dask.py 项目： snmz216/blaze

 def compute_broadcast(expr, *data, **kwargs):
     expr_inds = tuple(range(ndim(expr)))[::-1]
     func = get_numba_ufunc(expr)
     return atop(func,
                 expr_inds,
                 *concat(
                     (dat, tuple(range(ndim(dat))[::-1])) for dat in data),
                 dtype=data[-1].dtype)