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. tensor: The concated result. lengths: Output tensor containing sizes of the 0th dimension of tensors in the list, used for computing the gradient.
Nested Classes
class | TensorListConcat.Options | Optional attributes for TensorListConcat
|
Public Methods
static <T> TensorListConcat<T> |
create(Scope scope, Operand<?> inputHandle, Class<T> elementDtype, Options... options)
Factory method to create a class wrapping a new TensorListConcat operation.
|
static TensorListConcat.Options |
elementShape(Shape elementShape)
|
Output<Long> |
lengths()
|
Output<T> |
tensor()
|
Inherited Methods
Public Methods
public static TensorListConcat<T> create (Scope scope, Operand<?> inputHandle, Class<T> elementDtype, Options... options)
Factory method to create a class wrapping a new TensorListConcat operation.
Parameters
scope | current scope |
---|---|
options | carries optional attributes values |
Returns
- a new instance of TensorListConcat