def test_reshape_2():
    npr.seed(2)

    np_A = npr.randn(5,10)
    A    = kayak.Parameter(np_A)
    B    = kayak.Reshape(A, (2,25))
    C    = kayak.Parameter(npr.randn(25,5))
    D    = kayak.MatMult(B, C)
    out  = kayak.MatSum(D)

    out.value
    assert out.grad(A).shape == np_A.shape
    assert MAX_GRAD_DIFF < kayak.util.checkgrad(A, out)
dataset/ETHPy150Open HIPS/Kayak/tests/test_Reshape.py/test_reshape_2/swapped operands of `<`