search

eaplatanios / tensorflow_scala

TensorFlow API for the Scala Programming Language
http://platanios.org/tensorflow_scala/
Apache License 2.0
936 stars 96 forks source link

Support for QR, Cholesky and Linear Solve #84

Open mandar2812 opened 6 years ago

mandar2812 commented 6 years ago

The tensorflow API current supports QR, Cholesky and solving of Linear systems, it would be nice if there was a handle for them in the scala API as well.

eaplatanios commented 6 years ago

@mandar2812 Yeah it would indeed be good to support those. My API currently allows you to easily add support for them. For example, you can see how math ops are supported here and the equivalent Python code is here. Similar you could implement support for the linear algebra ops by porting this Python module (gradients implementation is here). It would be great to have that, but unfortunately I don't have time to implement this at the moment. I'm working on a machine translation library that I'll be releasing soon. Pull requests are always appreciated though. :)

I also wrote some brief instructions in #32 a while ago, that might be helpful.

mandar2812 commented 6 years ago

@eaplatanios Sure! Ill look into it when I have some time and update this thread.