search

quantalea / AleaTK

Library for general purpose numerical computing and Machine Learning based on tensors and tensor expressions.
http://www.aleatk.com
Apache License 2.0
75 stars 25 forks source link

Adding padding to the 2d convolutional layer #6

Open HunorSzabo opened 7 years ago

HunorSzabo commented 7 years ago

I started wondering looking at the code of the Convolutional layer that would it be so easy to add the padding to the descriptor. But after running a few tests and outputting the filters it looks good and also the dimension matches. Would it be so easy breeze?

soloman817 commented 7 years ago

Alea.cuDNN is just a P/Invoke wrapping over nvidia's cuDNN library,