bbba8ec7541528a288bbb833b02b2e0609e6de0f,t3f/riemannian.py,,project_sum,#Any#Any#Any#,10

Before Change

    return TensorTrainBatch(res_cores_list, where.get_raw_shape(),
                            batch_size=output_batch_size)
  else:
    return TensorTrain(res_cores_list, where.get_raw_shape())


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

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(what, where):
  Project `what` TTs on the tangent space of `where` TT.

Instances