dmlc / minerva

Minerva: a fast and flexible tool for deep learning on multi-GPU. It provides ndarray programming interface, just like Numpy. Python bindings and C++ bindings are both available. The resulting code can be run on CPU or GPU. Multi-GPU support is very easy.
Other
698 stars 172 forks source link

Request owl.elewise.pow/sqrt #42

Open CrabTY opened 9 years ago

CrabTY commented 9 years ago

Hi guys,

Is that possible for you to provide a element-wise power/sqrt function for the python interface? I'd like to adjust the weight use Adagrad or RMSprop, which requires a such operation. However, the numpy solution does not work well on multi-GPU case. Or can you show me some instruction about how to add such operations?

Thanks. Tao

jermainewang commented 9 years ago

Hi,

We will work on that. Currently it is still quite annoying to add new operators. If you wanna have a look. You could try trace the codes in:

Thanks for the feedback, Minjie