Python atopの例

コード例 #1

ファイル: dask.py プロジェクト: blaze/blaze

def compute_up(expr, data, **kwargs):
    leaf = expr._leaves()[0]
    chunk = symbol('chunk', DataShape(*(tuple(map(first, data.chunks)) +
                                        (leaf.dshape.measure,))))
    (chunk, chunk_expr), (agg, agg_expr) = split(expr._child, expr,
                                                 chunk=chunk)

    inds = tuple(range(ndim(leaf)))
    dtype = expr.dshape.measure.to_numpy_dtype()
    tmp = atop(
        curry(compute_it, chunk_expr, [chunk], **kwargs),
        inds,
        data,
        inds,
        dtype=dtype,
    )

    return atop(
        compose(
            curry(compute_it, agg_expr, [agg], **kwargs),
            curry(_concatenate2, axes=expr.axis),
        ),
        tuple(i for i in inds if i not in expr.axis),
        tmp,
        inds,
        dtype=dtype,
    )

コード例 #2

ファイル: dask.py プロジェクト: snmz216/blaze

def compute_up(expr, data, **kwargs):
    leaf = expr._leaves()[0]
    chunk = symbol(
        'chunk',
        DataShape(*(tuple(map(first, data.chunks)) + (leaf.dshape.measure, ))))
    (chunk, chunk_expr), (agg, agg_expr) = split(expr._child,
                                                 expr,
                                                 chunk=chunk)

    inds = tuple(range(ndim(leaf)))
    dtype = expr.dshape.measure.to_numpy_dtype()
    tmp = atop(
        curry(compute_it, chunk_expr, [chunk], **kwargs),
        inds,
        data,
        inds,
        dtype=dtype,
    )

    return atop(
        compose(
            curry(compute_it, agg_expr, [agg], **kwargs),
            curry(_concatenate2, axes=expr.axis),
        ),
        tuple(i for i in inds if i not in expr.axis),
        tmp,
        inds,
        dtype=dtype,
    )

コード例 #3

ファイル: dask.py プロジェクト: snmz216/blaze

def elemwise_array(expr, *data, **kwargs):
    leaves = expr._inputs
    expr_inds = tuple(range(ndim(expr)))[::-1]
    return atop(curry(compute_it, expr, leaves, **kwargs),
                expr_inds,
                *concat((dat, tuple(range(ndim(dat))[::-1])) for dat in data),
                dtype=expr.dshape.measure.to_numpy_dtype())

コード例 #4

ファイル: dask.py プロジェクト: Webs234/blaze

def elemwise_array(expr, *data, **kwargs):
    leaves = expr._inputs
    expr_inds = tuple(range(ndim(expr)))[::-1]
    return atop(
        curry(compute_it, expr, leaves, **kwargs),
        expr_inds,
        *concat((dat, tuple(range(ndim(dat))[::-1])) for dat in data)
    )

コード例 #5

ファイル: 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)

コード例 #6

ファイル: dask.py プロジェクト: blaze/blaze

def elemwise_array(expr, *data, **kwargs):
    leaves = expr._inputs
    expr_inds = tuple(range(ndim(expr)))[::-1]
    return atop(
        curry(compute_it, expr, leaves, **kwargs),
        expr_inds,
        *concat((dat, tuple(range(ndim(dat))[::-1])) for dat in data),
        dtype=expr.dshape.measure.to_numpy_dtype()
    )

コード例 #7

ファイル: dask.py プロジェクト: Webs234/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))

コード例 #8

ファイル: dask.py プロジェクト: somu-analyst/blaze

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