T
- data type for value()
output@Operator public final class TensorArrayConcat<T> extends PrimitiveOp
Takes `T` elements of shapes
(n0 x d0 x d1 x ...), (n1 x d0 x d1 x ...), ..., (n(T-1) x d0 x d1 x ...)
and concatenates them into a Tensor of shape:
(n0 + n1 + ... + n(T-1) x d0 x d1 x ...)
All elements must have the same shape (excepting the first dimension).Modifier and Type | Class and Description |
---|---|
static class |
TensorArrayConcat.Options
Optional attributes for
TensorArrayConcat |
operation
Modifier and Type | Method and Description |
---|---|
static <T> TensorArrayConcat<T> |
create(Scope scope,
Operand<?> handle,
Operand<Float> flowIn,
Class<T> dtype,
TensorArrayConcat.Options... options)
Factory method to create a class wrapping a new TensorArrayConcat operation.
|
static TensorArrayConcat.Options |
elementShapeExcept0(Shape elementShapeExcept0) |
Output<Long> |
lengths()
A vector of the row sizes of the original T elements in the
value output.
|
Output<T> |
value()
All of the elements in the TensorArray, concatenated along the first
axis.
|
equals, hashCode, op, toString
public static <T> TensorArrayConcat<T> create(Scope scope, Operand<?> handle, Operand<Float> flowIn, Class<T> dtype, TensorArrayConcat.Options... options)
scope
- current scopehandle
- The handle to a TensorArray.flowIn
- A float scalar that enforces proper chaining of operations.dtype
- The type of the elem that is returned.options
- carries optional attributes valuespublic static TensorArrayConcat.Options elementShapeExcept0(Shape elementShapeExcept0)
elementShapeExcept0
- The expected shape of an element, if known,
excluding the first dimension. Used to validate the shapes of
TensorArray elements. If this shape is not fully specified, concatenating
zero-size TensorArrays is an error.public Output<T> value()
Copyright © 2022. All rights reserved.