Function mul

imperative/python/test/unit/core/test_subgraph.py:129–135

(inputs, f, c)

        custom_grad=True,
    )
    def mul(inputs, f, c):
        x, y = inputs[0:2]
        z = f("*", x, y)
        (dz,) = yield (z,)
        dx = f("*", dz, y)
        dy = f("*", dz, x)
        yield (dx, dy)

    return mul