testSqueeze :: Test
testSqueeze =
  testCase "testSqueeze" $ do
    ([dx], [s]) <-
      TF.runSession $ do
        (x :: TF.Tensor TF.Value Float) <- TF.render $ TF.zeros $ TF.Shape [1, 2, 3 :: Int64]
        let y = TF.squeeze x
        calculateGradWithShape y x
    V.fromList [1, 1, 1, 1, 1, 1] @=? dx
    V.fromList [1, 2, 3] @=? s
testSqueeze :: Test
testSqueeze =
  testCase "testSqueeze" $ do
    ([dx], [s]) <-
      TF.runSession $ do
        (x :: TF.Tensor TF.Value Float) <- TF.render $ TF.zeros $ TF.Shape [1, 2, 3 :: Int64]
        let y = TF.squeeze x
        calculateGradWithShape y x
    V.fromList [1, 1, 1, 1, 1, 1] @=? dx
    V.fromList [1, 2, 3] @=? s
testSqueeze =
  testCase "testSqueeze" $ do
    ([dx], [s]) <-
      TF.runSession $ do
        (x :: TF.Tensor TF.Value Float) <- TF.render $ TF.zeros $ TF.Shape [1, 2, 3 :: Int64]
        let y = TF.squeeze x
        calculateGradWithShape y x
    V.fromList [1, 1, 1, 1, 1, 1] @=? dx
    V.fromList [1, 2, 3] @=? s