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.


class Conv: Neural network layer for 2D convolution.

class Deconv: Neural network layer for 2D transposed convolution.


conv_info(...): Returns parameters and output shape information given input shapes.