conv_transpose2d

Function conv_transpose2d 

Source
pub fn conv_transpose2d<R: CubeRuntime>(
    input: CubeTensor<R>,
    weight: CubeTensor<R>,
    bias: Option<CubeTensor<R>>,
    options: ConvTransposeOptions<2>,
    strategy: ConvTranspose2dStrategy,
) -> Result<CubeTensor<R>, ConvSetupError>
Expand description

Performs 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.