search

arrayfire / arrayfire-ml

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

[WIP] Conv2D Layer #39

Open plavin opened 6 years ago

plavin commented 6 years ago

This still isn't quite perfect. The only thing left to do is to figure out how to make grad_func work with batched output.