lmjohns3 / theanets

Neural network toolkit for Python
http://theanets.rtfd.org
MIT License
328 stars 74 forks source link

use shape information in layers where possible #126

Closed lmjohns3 closed 8 years ago

lmjohns3 commented 8 years ago

Currently network layers know only their size, which is basically the number of channels or variables in their outputs. For feedforward and most recurrent models this is sufficient, but for convolution models this quickly becomes painful to deal with since the user needs to compute the sizes of the image patches at each step. theanets should permit each layer to know its shape so that we can simplify some of this for the end user.