ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: oryx.experimental.nn.convolution

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.