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>
tensor()

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 ()