How to convert NCHW to NHWC in tvm?

Questions
6

Hi @srkreddy1238 and @joshherr can I use?

inline Tensor transpose(const Tensor& x,
                        Array<Integer> axes,
                        std::string name = "tensor",
                        std::string tag = kInjective)

from topi\include\topi\transform.h header ?