78714246f7c410fcbe26f1e24bf313ac97e88946,ops_test.py,TTTensorTest,testAdd,#TTTensorTest#,80

Before Change

    with self.test_session() as sess:
      res_actual = ops.full(ops.add(tt_a, tt_b))
      res_desired = ops.full(tt_a) + ops.full(tt_b)
      res_actual_val, res_desired_val = sess.run([res_actual, res_desired])
      self.assertAllClose(res_actual_val, res_desired_val)

  def testMultiply(self):

After Change

    tt_b = initializers.random_tensor((2, 1, 3, 4), tt_rank=[1, 2, 4, 3, 1])
    with self.test_session() as sess:
      res_actual = ops.full(ops.add(tt_a, tt_b))
      res_actual2 = ops.full(tt_a + tt_b)
      res_desired = ops.full(tt_a) + ops.full(tt_b)
      to_run = [res_actual, res_actual2, res_desired]
      res_actual_val, res_actual2_val, res_desired_val = sess.run(to_run)
      self.assertAllClose(res_actual_val, res_desired_val)
      self.assertAllClose(res_actual2_val, res_desired_val)

Instances