

def test_roll_no_axis():

    def fun(x):
        return to_scalar(np.roll(x, 2, axis=1))
    d_fun = (lambda x: to_scalar(grad(fun)(x)))
    mat = npr.randn(4, 5)
    check_grads(fun, mat)
    check_grads(d_fun, x)

dataset/ETHPy150Open HIPS/autograd/tests/test_numpy.py test_roll_no_axis/VarMisuse@76/90 `mat`->`x`