U
- data type for tensor()
output@Operator public final class TensorListConcatV2<U> extends PrimitiveOp
Requires that all tensors have the same shape except the first dimension.
input_handle: The input list. element_shape: The shape of the uninitialized elements in the list. If the first dimension is not -1, it is assumed that all list elements have the same leading dim. leading_dims: The list of leading dims of uninitialized list elements. Used if the leading dim of input_handle.element_shape or the element_shape input arg is not already set. tensor: The concated result. lengths: Output tensor containing sizes of the 0th dimension of tensors in the list, used for computing the gradient.
operation
Modifier and Type | Method and Description |
---|---|
static <U,T extends Number> |
create(Scope scope,
Operand<?> inputHandle,
Operand<T> elementShape,
Operand<Long> leadingDims,
Class<U> elementDtype)
Factory method to create a class wrapping a new TensorListConcatV2 operation.
|
Output<Long> |
lengths() |
Output<U> |
tensor() |
equals, hashCode, op, toString
public static <U,T extends Number> TensorListConcatV2<U> create(Scope scope, Operand<?> inputHandle, Operand<T> elementShape, Operand<Long> leadingDims, Class<U> elementDtype)
scope
- current scopeinputHandle
- elementShape
- leadingDims
- elementDtype
- Copyright © 2022. All rights reserved.