Bihaqo / t3f

Tensor Train decomposition on TensorFlow
https://t3f.readthedocs.io/en/latest/index.html
MIT License
219 stars 55 forks source link

Batch and broadcasting support for t3f.ops.multiply #75

Closed izmailovpavel closed 6 years ago

izmailovpavel commented 7 years ago

It looks like t3f.ops.multiply doesn't support batches. It would be nice if it could also broadcast if we need to multiply a tt-batch by a single tt-matrix.

Bihaqo commented 6 years ago

91 Done