Каков наилучший способ преобразовать тензор из формата NHWC в формат NCHW и наоборот?

Есть ли конкретная операция, которая делает это, или мне нужно будет использовать некоторую комбинацию операций типа split / concat?