pub fn conv_forward<R: CubeRuntime, const N: usize>(
input: CubeTensor<R>,
weight: CubeTensor<R>,
bias: Option<CubeTensor<R>>,
options: ConvOptions<N>,
strategy: ConvStrategy,
) -> Result<CubeTensor<R>, ConvSetupError>Expand description
Performs an N-dimensional convolution with the given strategy
input- The input feature mapweight- The weights (filter) applied to each kernelbias- The bias added to each channeloptions- The options to use for the convolutionstrategy- The convolution algorithm to use. Autotune will pick the fastest available option.