Attend the Women in ML Symposium on December 7 Register now

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

public static TensorListConcat.Options elementShape (Shape elementShape)

public Output<Long> lengths ()

public Output<T> tensor ()