4c57c23e138e8b26427337d67536d4e239d55e06,tensor_train_test.py,TensorTrainTest,testValidateTTCores2d,#TensorTrainTest#,11

Before Change

                ((1, 2, 1, 1), False))

    for ranks, desired in schedule:
      a = np.random.rand(ranks[0], 10, ranks[1]).astype(np.float32)
      b = np.random.rand(ranks[2], 9, ranks[3]).astype(np.float32)
      with self.test_session():
        self.assertEqual(desired, tensor_train._are_tt_cores_valid((a, b)))
        if desired:
          tf_tens = tensor_train.TensorTrain((a, b))
          tf_cores = tf_tens.tt_cores
          self.assertEqual(desired, tensor_train._are_tt_cores_valid(tf_cores))
        else:
          with self.assertRaises(ValueError):
            tensor_train.TensorTrain((a, b))

After Change

      with self.test_session():
        actual = tensor_train._are_tt_cores_valid((a, b), (10, 9),
                                                  claimed_tt_ranks)
        self.assertEqual(desired, actual)
        // Wrong shape.
        actual = tensor_train._are_tt_cores_valid((a, b), (9, 9),
                                                  claimed_tt_ranks)
        self.assertEqual(False, actual)

Instances