Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

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