b8be1ed0e949d11310fbea50730d470ad2bb7859,t3f/ops_test.py,TTTensorTest,testCastIntFloat,#TTTensorTest#,137

Before Change

    K_3 = np.random.randint(0, high=100, size=(2, 2, 1))
    tt_int = TensorTrain([K_1, K_2, K_3], tt_ranks=[1, 2, 2, 1])
    
    for dtype in [tf.float16, tf.float32, tf.float64]:
      self.assertEqual(ops.cast(tt_int, dtype).dtype, dtype)


class TTMatrixTest(tf.test.TestCase):

  def testFullMatrix2d(self):
    np.random.seed(1)

After Change

    K_3 = np.random.randint(0, high=100, size=(2, 2, 1))
    tt_int = TensorTrain([K_1, K_2, K_3], tt_ranks=[1, 2, 2, 1])

    with self.test_session() as sess:
      for dtype in [tf.float16, tf.float32, tf.float64]:
        casted = ops.cast(tt_int, dtype)
        casted_val = sess.run(ops.full(casted))
        self.assertEqual(dtype, casted.dtype)
        self.assertTrue(dtype, casted_val.dtype)


class TTMatrixTest(tf.test.TestCase):

  def testFullMatrix2d(self):
    np.random.seed(1)

Instances