def nchw_to_nhwc(array):
	"""
	ONNX uses NCHW. ELINA expects NHWC

	:param array: array to be converted

	:return: converted array
	"""
	if array.ndim == 4:
		return array.transpose(0, 2, 3, 1)

	return array