

def test_min_axis_keepdims():

    def fun(x):
        return to_scalar(np.min(x, axis=1, keepdims=True))
    d_fun = (lambda x: to_scalar(grad(fun)(mat)))
    mat = npr.randn(3, 4, 5)
    check_grads(fun, mat)
    check_grads(d_fun, mat)

dataset/ETHPy150Open HIPS/autograd/tests/test_numpy.py test_min_axis_keepdims/VarMisuse@51/62 `x`->`mat`