View source on GitHub |
Like tf.nn.conv2d
except applies batch of kernels to batch of x
.
tfp.experimental.nn.util.make_convolution_fn(
filter_shape,
rank,
strides,
padding,
dilations=None,
dtype=tf.int32,
validate_args=False,
name=None
)