tensorflow:: ops:: DataFormatVecPermute
#include <nn_ops.h>
Permute input tensor from src_format
to dst_format
.
Summary
Given source and destination format strings of length n=4 or 5, the input tensor must be a vector of size n or n-2, or a 2D tensor of shape (n, 2) or (n-2, 2).
If the first dimension of the input tensor is n-2, it is assumed that non-spatial dimensions are omitted (i.e N
, C
).
For example, with src_format
of NHWC
, dst_format
of NCHW
, and input:
[1, 2, 3, 4], the output will be:
[1, 4, 2, 3]With
src_format
of NDHWC
, dst_format
of NCDHW
, and input: [[1, 6], [2, 7], [3, 8], [4, 9], [5, 10]], the output will be:
[[1, 6], [5, 10], [2, 7], [3, 8], [4, 9]]With
src_format
of NHWC
, dst_format
of NCHW
, and input: [1, 2], the output will be:
[1, 2]
Args:
Optional attributes (see Attrs
):
- src_format: source data format.
- dst_format: destination data format.
Returns:
Constructors and Destructors |
|
---|---|
DataFormatVecPermute(const ::tensorflow::Scope & scope, ::tensorflow::Input x)
|
|
DataFormatVecPermute(const ::tensorflow::Scope & scope, ::tensorflow::Input x, const DataFormatVecPermute::Attrs & attrs)
|
Public attributes |
|
---|---|
operation
|
|
y
|
Public functions |
|
---|---|
node() const
|
::tensorflow::Node *
|
operator::tensorflow::Input() const
|
|
operator::tensorflow::Output() const
|
|
Public static functions |
|
---|---|
DstFormat(StringPiece x)
|
|
SrcFormat(StringPiece x)
|
Structs |
|
---|---|
tensorflow:: |
Optional attribute setters for DataFormatVecPermute. |
Public attributes
operation
Operation operation
y
::tensorflow::Output y
Public functions
DataFormatVecPermute
DataFormatVecPermute( const ::tensorflow::Scope & scope, ::tensorflow::Input x )
DataFormatVecPermute
DataFormatVecPermute( const ::tensorflow::Scope & scope, ::tensorflow::Input x, const DataFormatVecPermute::Attrs & attrs )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
Public static functions
DstFormat
Attrs DstFormat( StringPiece x )
SrcFormat
Attrs SrcFormat( StringPiece x )