T
- data type for output()
output@Operator public final class Stack<T> extends PrimitiveOp implements Operand<T>
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`.Modifier and Type | Class and Description |
---|---|
static class |
Stack.Options
Optional attributes for
Stack |
operation
Modifier and Type | Method and Description |
---|---|
Output<T> |
asOutput()
Returns the symbolic handle of a tensor.
|
static Stack.Options |
axis(Long axis) |
static <T> Stack<T> |
create(Scope scope,
Iterable<Operand<T>> values,
Stack.Options... options)
Factory method to create a class wrapping a new Stack operation.
|
Output<T> |
output()
The packed tensor.
|
equals, hashCode, op, toString
public static <T> Stack<T> create(Scope scope, Iterable<Operand<T>> values, Stack.Options... options)
scope
- current scopevalues
- Must be of same shape and type.options
- carries optional attributes valuespublic static Stack.Options axis(Long axis)
axis
- Dimension along which to pack. Negative values wrap around, so the
valid range is `[-(R+1), R+1)`.public Output<T> asOutput()
Operand
Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
asOutput
in interface Operand<T>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.