Module conv 

Module with convolution operations.

Functions

calculate_conv_output_shape

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

calculate_conv_output_size

Calculate the expected output size when doing a convolution operation.

calculate_conv_output_sizes

Calculate the expected output sizes when doing a convolution operation.

calculate_conv_padding

Calculate the expected padding size required when applying a convolution.

calculate_conv_transpose_output_shape

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

calculate_conv_transpose_output_size

Calculate the expected output size when doing a transposed convolution operation.

calculate_pool_output_shape

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

calculate_pool_output_size

Calculate the expected output size when doing a pooling operation.