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]