search

josephjaspers / blackcat_tensors

Matrix-Vector Library Designed for Neural Network Construction. cuda (gpu) support, openmp (multithreaded cpu) support, partial support of BLAS, expression template based implementation PTX code generation identical to hand written kernels, and support for auto-differentiation
12 stars 4 forks source link

Convolution error #59

Open josephjaspers opened 4 years ago

josephjaspers commented 4 years ago

Convolution does not go be the correct results when channels != 1

Fix includes a rewrite if the current image to col operation

josephjaspers commented 4 years ago

FIXED!

TODO

add support for when strides != 1 and/or dilation != 1