calculate_conv_output_shape

burn::tensor::backend::ops::conv

Function calculate_conv_output_shape 

pub fn calculate_conv_output_shape<const N: usize>(
    in_shape: &Shape,
    weight_shape: &Shape,
    stride: &[usize; N],
    padding: &[usize; N],
    dilation: &[usize; N],
) -> Result<Shape, ShapeError>
Expand description

Calculate the expected output shape [batch_size, channels_out, spatial_dims, ..] for a convolution.