

def test_squeeze_func():
    A = npr.randn(5, 1, 4)

    def fun(x):
        return to_scalar(np.squeeze(x))
    d_fun = (lambda x: to_scalar(grad(fun)(x)))
    check_grads(fun, A)
    check_grads(A, A)

dataset/ETHPy150Open HIPS/autograd/tests/test_numpy.py test_squeeze_func/VarMisuse@70/82 `d_fun`->`A`