riemannian projection of two tensor train objects fails

Bihaqo / t3f

Tensor Train decomposition on TensorFlow

https://t3f.readthedocs.io/en/latest/index.html

MIT License

219 stars 55 forks source link

riemannian projection of two tensor train objects fails #80

Closed KhrulkovV closed 6 years ago

KhrulkovV commented 6 years ago

The following code projected_grad = t3f.riemannian.project(grad_tt, weight_tt) where grad_tt and weight_tt are TensorTrain objects gives the following error ValueError: batch size is not available for a TensorTrain object.

However, the same piece of code with t3f.riemannian.project_sum(grad_tt, weight_tt) instead works

Bihaqo commented 6 years ago

Works in the latest version (develop branch)