ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

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 .

Args:

  • 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:

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
)