conv_forward

Function conv_forward 

Source
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 map
  • weight - The weights (filter) applied to each kernel
  • bias - The bias added to each channel
  • options - The options to use for the convolution
  • strategy - The convolution algorithm to use. Autotune will pick the fastest available option.