Conv:
o = output
p = padding
k = kernel_size
s = stride
d = dilation
o = [i + 2*p - k - (k-1)*(d-1)]/s + 1
ConvTranspose:
o = (i -1)*s - 2*p + k + output_padding
Conv:
o = output
p = padding
k = kernel_size
s = stride
d = dilation
o = [i + 2*p - k - (k-1)*(d-1)]/s + 1
ConvTranspose:
o = (i -1)*s - 2*p + k + output_padding