bbba8ec7541528a288bbb833b02b2e0609e6de0f,t3f/riemannian.py,,project,#Any#Any#,181

Before Change

    res_cores_list.append(res_core)
  // TODO: TT-ranks.
  if output_is_batch:
    return TensorTrainBatch(res_cores_list, where.get_raw_shape(),
                            batch_size=output_batch_size)
  else:
    return TensorTrain(res_cores_list, where.get_raw_shape())

After Change

    res_cores_list.append(res_core)
  // TODO: TT-ranks.
  if output_is_batch:
    res = TensorTrainBatch(res_cores_list, where.get_raw_shape(),
                            batch_size=output_batch_size)
  else:
    res = TensorTrain(res_cores_list, where.get_raw_shape())

  res.projection_on = where
  return res


def project_matmul(what, where, matrix):
  Project `matrix` * `what` TTs on the tangent space of `where` TT.

Instances