NN_nhwc_to_nchw()

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

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

Parameters

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