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. 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>
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.
Output < TInt64 >
Output <U>

Inherited Methods

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "TensorListConcatV2"

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

public Output < TInt64 > lengths ()

public Output <U> tensor ()