tf.raw_ops.DataFormatVecPermute

Переупорядочивает входной тензор из src_format в dst_format.

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

Учитывая строки формата источника и назначения длиной n=4 или 5, входной тензор должен быть вектором размера n или n-2, или двумерным тензором формы (n, 2) или (n-2, 2).

Если первая размерность входного тензора равна n-2, предполагается, что не пространственные размерности опущены (т.е. N, C).

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

[1, 2, 3, 4]

, выход будет:

[1, 4, 2, 3]

С src_format формата NDHWC, dst_format формата NCDHW и входом:

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

, выход будет:

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

С src_format формата NHWC, dst_format формата NCHW и входом:

[1, 2]

, выход будет:

[1, 2]