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

Before Change

    res_ranks = []
    for core_idx in range(ndims + 1):
      res_ranks.append(a_ranks[core_idx] * b_ranks[core_idx])
    res_ranks = tf.TensorShape(res_ranks)
  else:
    res_ranks = None
  return TensorTrain(result_cores, res_shape, res_ranks)

After Change

  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