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?
Awesome method! Why not combine permute and transpose or make transpose inaccessible to user since it’s used internally by permute as mentioned by fmassa.