tf.raw_ops.DataFormatVecPermute

Переставляет входной тензор из src_format в dst_format.

tf.raw_ops.DataFormatVecPermute(
    x, src_format='NHWC', dst_format='NCHW', name=None
)

Входной тензор должен быть вектором размера 4 или тензором 4x2.

Например, при src_format формате NHWC, dst_format формате NCHW, и входных данных:

[1, 2, 3, 4]

и

[[1, 2, 3, 4],
 [5, 6, 7, 8]]

, выходными данными будут (соответственно):

[1, 4, 2, 3]

и

[[1, 4, 2, 3],
 [5, 8, 6, 7]]