tensorflow::ops::Stack
#include <array_ops.h>
Packs a list of N
rank-R
tensors into one rank-(R+1)
tensor.
Summary
Packs the N
tensors in values
into a tensor with rank one higher than each tensor in values
, by packing them along the axis
dimension. Given a list of tensors of shape (A, B, C)
;
if axis == 0
then the output
tensor will have the shape (N, A, B, C)
. if axis == 1
then the output
tensor will have the shape (A, N, B, C)
. Etc.
For example:
# 'x' is [1, 4] # 'y' is [2, 5] # 'z' is [3, 6] pack([x, y, z]) => [[1, 4], [2, 5], [3, 6]] # Pack along first dim. pack([x, y, z], axis=1) => [[1, 2, 3], [4, 5, 6]]
This is the opposite of unpack
.
Arguments:
- scope: A Scope object
- values: Must be of same shape and type.
Optional attributes (see Attrs
):
- axis: Dimension along which to pack. Negative values wrap around, so the valid range is
[-(R+1), R+1)
.
Returns:
-
Output
: The packed tensor.
Constructors and Destructors | |
---|---|
Stack(const ::tensorflow::Scope & scope, ::tensorflow::InputList values) | |
Stack(const ::tensorflow::Scope & scope, ::tensorflow::InputList values, const Stack::Attrs & attrs) |
Public attributes | |
---|---|
operation | |
output |
Public functions | |
---|---|
node() const | ::tensorflow::Node * |
operator::tensorflow::Input() const | |
operator::tensorflow::Output() const |
Public static functions | |
---|---|
Axis(int64 x) |
Structs | |
---|---|
tensorflow::ops::Stack::Attrs | Optional attribute setters for Stack. |
Public attributes
operation
Operation operation
output
::tensorflow::Output output
Public functions
Stack
Stack( const ::tensorflow::Scope & scope, ::tensorflow::InputList values )
Stack
Stack( const ::tensorflow::Scope & scope, ::tensorflow::InputList values, const Stack::Attrs & attrs )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
Public static functions
Axis
Attrs Axis( int64 x )
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/cc/class/tensorflow/ops/stack