Конвертировать нейросеть из NCHW в NHWC

Имеется сеть в формате onnx, выходные данные слоев которой сейчас в формате NCHW (батч-канал-высота-ширина). Требуется сконвертировать сеть таким образом, чтобы все внутренние тензоры были в формате NHWC.Сам файл onnx не могу здесь загрузить, по запросу