arrayfire / arrayfire-ml

ArrayFire's Machine Learning Library.
BSD 3-Clause "New" or "Revised" License
102 stars 23 forks source link

Add ELU, PReLU, ThresholdedReLU layers #32

Closed plavin closed 7 years ago

plavin commented 7 years ago

This is based off of the Keras documentation here The PReLU layer in Keras supports more features that we can discuss adding in the future. (e.g. weight initialization function, regularization).