conv2dOutputShape
fun conv2dOutputShape(inputShape: IntArray, weightShape: IntArray, stride: Pair<Int, Int>, padding: Pair<Int, Int>, dilation: Pair<Int, Int>): IntArray(source)
Conv2d output shape: input (batch, in_channels, height, width), weight (out_channels, in_channels_per_group, kernel_h, kernel_w), result (batch, out_channels, out_h, out_w).