Module convolution 

Convolution operations for batched data

Functions

conv2d_batch

2D convolution for batched images Input shape: [batch_size, in_channels, height, width] Weight shape: [out_channels, in_channels, kernel_height, kernel_width] Output shape: [batch_size, out_channels, out_height, out_width]