Concats all tensors in the list along the 0th dimension.
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.
Constants
String | OP_NAME | The name of this op, as known by TensorFlow core engine |
Public Methods
static <U extends TType> TensorListConcat<U> | |
Output<TInt64> |
lengths()
|
Output<U> |
tensor()
|
Inherited Methods
Constants
public static final String OP_NAME
The name of this op, as known by TensorFlow core engine
Public Methods
public static TensorListConcat<U> create (Scope scope, Operand<?> inputHandle, Operand<? extends TNumber> elementShape, Operand<TInt64> leadingDims, Class<U> elementDtype)
Factory method to create a class wrapping a new TensorListConcat operation.
Parameters
scope | current scope |
---|
Returns
- a new instance of TensorListConcat