search

autumnai / leaf

Open Machine Intelligence Framework for Hackers. (GPU/CPU)
Apache License 2.0
5.56k stars 270 forks source link

Support convolutions with 3(+) spatial dimensions #61

Open hobofan opened 8 years ago

hobofan commented 8 years ago

The convolution layer currently only supports 2D convolutions.

Supporting 3D and higher dimensional convolutions might just require relaxing the num_spatial_dims bounds to work, but there also needs to be some testing for that.