NN_nchw_to_nhwc()

void NN_nchw_to_nhwc	(	Tensor *	out,
		const Tensor *	in
	)

Converts a tensor from NCHW (PyTorch) to NHWC (Channel-last) format.

Parameters

out	the output tensor of shape (batch_size, height, width, channels)
in	the input tensor of shape (batch_size, channels, height, width)