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

tfp.experimental.joint_distribution_layers.make_conv2d_layer_class

Creates a Conv2DLayer class.

strides A 2-tuple of positive integers. Strides for the spatial dimensions.
padding A Python string. Can be either 'SAME' or 'VALID'.

conv2d_layer_class A new Conv2DLayer class that closes over the args to this function.