Convolution Functions - Githubissues

arrayfire / arrayfire-ml

ArrayFire's Machine Learning Library.

BSD 3-Clause "New" or "Revised" License

102 stars 23 forks source link

Convolution Functions #33

Open pavanky opened 7 years ago

pavanky commented 7 years ago

The following functions and modules need to be implemented:

[ ] Basic Convolutions: No strides, No pooling
[ ] Strided Convolutions without pooling
[ ] Strided Convolutions with pooling

Merging Strided Convolutions with pooling could potentially allow call wrap and unwrap to be only called once instead of twice.

MichaelMcCulloch commented 2 years ago

Hello, I see this thread has been silent for quite some time, however I have need of a way to implement max pooling with arrayfire, and I was interested to know how you planned to do it?