burn_jit::kernel::conv

Function conv_transpose2d

Source 
pub fn conv_transpose2d<R: JitRuntime, E: FloatElement, I: IntElement>(
    input: JitTensor<R>,
    weight: JitTensor<R>,
    bias: Option<JitTensor<R>>,
    options: ConvTransposeOptions<2>,
    strategy: ConvTranspose2dStrategy,
) -> Result<JitTensor<R>, ConvLaunchError>
Expand description

Perform a 2D 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.