Pytorch:How to change weights layout from NCHW to NHWC

[fush]

My device only supports this layout, so I must have to change it.