def testIssue776(self):
    'Tests that the scatter-add transpose rule instantiates symbolic zeros.'

    def f(u):
        y = jax.ops.index_add(onp.ones(10), [2, 4, 5], u)
        return lax.tie_in(y, 7.0)
    self.assertAllClose(onp.zeros(3), api.grad(f)(onp.ones(3)), check_dtypes=True)
Tests that the scatter-add transpose rule instantiates symbolic zeros.
def testIssue776(self):
    

    def f(u):
        y = jax.ops.index_add(onp.ones(10), [2, 4, 5], u)
        return lax.tie_in(y, 7.0)
    self.assertAllClose(onp.zeros(3), api.grad(f)(onp.ones(3)), check_dtypes=True)