conv2d_optimized

Function conv2d_optimized 

Source
pub fn conv2d_optimized(
    input: &Tensor,
    weight: &Tensor,
    bias: Option<&Tensor>,
    stride: (usize, usize),
    padding: (usize, usize),
) -> Result<Tensor, GhostError>
Expand description

Optimized 2D convolution