View source on GitHub |
Creates a Conv2DLayer
class.
tfp.experimental.joint_distribution_layers.make_conv2d_layer_class(
strides, padding
)
Args | |
---|---|
strides
|
A 2-tuple of positive integers. Strides for the spatial dimensions. |
padding
|
A Python string. Can be either 'SAME' or 'VALID'. |
Returns | |
---|---|
conv2d_layer_class
|
A new Conv2DLayer class that closes over the args to
this function.
|