Pytorch:How to change weights layout from NCHW to NHWC

Thank you for answering my questions.It’s very useful and answers my doubts.

I have another question .I have my own data layout,the weights layout is not the NHWC but the (N/16)HW(16n)C.

How can I customize the data layout?

Is there a document link to learn how to customize the data layout?

Thanks.