mabor::tensor::ops::conv

Function calculate_conv_output_sizes

pub fn calculate_conv_output_sizes(
    kernel_size: &[usize],
    stride: &[usize],
    padding: &[usize],
    dilation: &[usize],
    size_in: &[usize],
) -> Vec<usize>
Expand description

Calculate the expected output sizes when doing a convolution operation.