cff7c0087b9aae064aa7cd08de67a76ea19c6e63,ops.py,,tt_tt_matmul,#Any#Any#,46

Before Change

  a_shape = shapes.lazy_raw_shape(tt_matrix_a)
  tt_ranks_defined = True
  if tt_matrix_a.get_tt_ranks().is_fully_defined():
    a_ranks = tt_matrix_a.get_tt_ranks().as_list()
  else:
    a_ranks = shapes.tt_ranks(tt_matrix_a)
    tt_ranks_defined = False
  b_shape = shapes.lazy_raw_shape(tt_matrix_b)

After Change

  res_shape = (tt_matrix_a.get_raw_shape()[0], tt_matrix_b.get_raw_shape()[1])
  static_a_ranks = tt_matrix_a.get_tt_ranks()
  static_b_ranks = tt_matrix_b.get_tt_ranks()
  out_ranks = [a_r * b_r for a_r, b_r in zip(static_a_ranks, static_b_ranks)]
  return TensorTrain(result_cores, res_shape, tf.TensorShape(out_ranks))


def tt_dense_matmul(tt_matrix_a, matrix_b):

Instances