General n-dimensional convolution operator, with optional dilation.
The semantics of this operation mimic the jax.lax.conv_general_dilated function in JAX, which wraps XLA's general convolution operator.
jax.lax.conv_general_dilated
Grouped convolutions are not supported right now.
General n-dimensional convolution operator, with optional dilation.
The semantics of this operation mimic the
jax.lax.conv_general_dilatedfunction in JAX, which wraps XLA's general convolution operator.Grouped convolutions are not supported right now.