description: Permute input tensor from src_format to dst_format.
robots: noindex
Permute input tensor from src_format to dst_format.
tf.raw_ops.DataFormatVecPermute(
x, src_format='NHWC', dst_format='NCHW', name=None
)
Input tensor must be a vector of size 4, or a 4x2 tensor.
For example, with src_format of NHWC, dst_format of NCHW, and inputs:
[1, 2, 3, 4]
and
[[1, 2, 3, 4],
[5, 6, 7, 8]]
, the outputs will be (respectively):
[1, 4, 2, 3]
and
[[1, 4, 2, 3],
[5, 8, 6, 7]]
Args | |
|---|---|
x
|
A Tensor. Must be one of the following types: int32, int64.
Vector of size 4 or Tensor of shape (4, 2) in source data format.
|
src_format
|
An optional string. Defaults to "NHWC".
source data format.
|
dst_format
|
An optional string. Defaults to "NCHW".
destination data format.
|
name
|
A name for the operation (optional). |
Returns | |
|---|---|
A Tensor. Has the same type as x.
|