View source on GitHub |
Contains building blocks for convolutional neural networks.
Classes
class Conv
: Neural network layer for 2D convolution.
class Deconv
: Neural network layer for 2D transposed convolution.
Functions
conv_info(...)
: Returns parameters and output shape information given input shapes.