Hi, in tensorflow, we have data_format option in tf.nn.conv2d which could specify the data format as NHWC or NCHW.
Is there equivalent operation in pytorch?
If not, should we convert Variable to numpy.array, use np.swapaxes and convert it back into Variable?
And under such circumstances, will the gradient tracked properly?