

def test_mean_3():

    def fun(x):
        return to_scalar(np.mean(x, axis=0, keepdims=True))
    d_fun = (lambda x: to_scalar(grad(fun)(x)))
    mat = npr.randn(10, 11)
    check_grads(fun, x)
    check_grads(d_fun, mat)

dataset/ETHPy150Open HIPS/autograd/tests/test_numpy.py test_mean_3/VarMisuse@71/82 `mat`->`x`