Solved by phan_phan in post #2 
              
              
                So, you have filters in a tensor w of shape: 
[batch_size, out_channels, in_channels, kernel_height, kernel_width], 
and images in a tensor x of shape: 
[batch_size, in_channels, in_height, in_width], 
and you want an output of shape: 
[batch_size, out_channels, out_height, out_width], 
where the i_…