TensorListConcat

public final class TensorListConcat

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>
Output <T>

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

public static TensorListConcat.Options elementShape ( Shape elementShape)

public Output <Long> lengths ()

public Output <T> tensor ()