Function dfdx::tensor_ops::conv2d

source · [−]

pub fn conv2d<TAPE: 'static + Tape, const IN_CHAN: usize, const OUT_CHAN: usize, const KERNEL: usize, const STRIDE: usize, const PADDING: usize, const IN_HEIGHT: usize, const IN_WIDTH: usize>(
    x: Tensor3D<IN_CHAN, IN_HEIGHT, IN_WIDTH, TAPE>,
    filters: &Tensor4D<OUT_CHAN, IN_CHAN, KERNEL, KERNEL>,
    bias: &Tensor1D<OUT_CHAN>
) -> Tensor3D<OUT_CHAN, { _ }, { _ }, TAPE>

Expand description

Requires Nightly Perform a 2d convolution.

TODO docstring