testMeanGradient :: Test
testMeanGradient = testCase "testMeanGradient" $ do
    [dx] <- TF.runSession $ do
        x <- TF.render $ TF.vector [1, 2, 0 :: Float]
        let y = TF.mean x (TF.vector [0 :: Int32])
        TF.gradients y [x] >>= TF.run
    V.fromList [1, 1, 1 :: Float] @=? dx
testMeanGradient :: Test
testMeanGradient = testCase "testMeanGradient" $ do
    [dx] <- TF.runSession $ do
        x <- TF.render $ TF.vector [1, 2, 0 :: Float]
        let y = TF.mean x (TF.vector [0 :: Int32])
        TF.gradients y [x] >>= TF.run
    V.fromList [1, 1, 1 :: Float] @=? dx
testMeanGradient = testCase "testMeanGradient" $ do
    [dx] <- TF.runSession $ do
        x <- TF.render $ TF.vector [1, 2, 0 :: Float]
        let y = TF.mean x (TF.vector [0 :: Int32])
        TF.gradients y [x] >>= TF.run
    V.fromList [1, 1, 1 :: Float] @=? dx