Function _elwise

imperative/python/test/unit/core/test_autodiff.py:22–29

(mode)

def _elwise(mode):
    op = Elemwise(mode)

    def f(*args):
        (result,) = apply(op, *args)
        return result

    return f


add = _elwise(Elemwise.Mode.ADD)