tensorflow:: ops:: DataFormatVecPermute
#include <nn_ops.h>
Permute input tensor from src_format
to dst_format
.
Summary
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]]
Arguments:
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 )