Function dfdx::tensor_ops::conv2d_batched

source · [−]

pub fn conv2d_batched<TAPE: 'static + Tape, const BATCH_SIZE: usize, 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: Tensor4D<BATCH_SIZE, IN_CHAN, IN_HEIGHT, IN_WIDTH, TAPE>,
    filters: &Tensor4D<OUT_CHAN, IN_CHAN, KERNEL, KERNEL>,
    bias: &Tensor1D<OUT_CHAN>
) -> Tensor4D<BATCH_SIZE, OUT_CHAN, { _ }, { _ }, TAPE>

Expand description

Requires Nightly Perform a batched 2d convolution

TODO docstring