org.tensorflow.op.core

Class TensorListConcat<T>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.TensorListConcat<T>

Type Parameters:

T - data type for tensor() output

All Implemented Interfaces:

Op

@Operator
public final class TensorListConcat<T>
extends PrimitiveOp

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.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
static <T> TensorListConcat<T>	create(Scope scope, Operand<?> inputHandle, Class<T> elementDtype) Factory method to create a class to wrap a new TensorListConcat operation to the graph.
Output<Long>	lengths()
Output<T>	tensor()

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

public static <T> TensorListConcat<T> create(Scope scope,
                                             Operand<?> inputHandle,
                                             Class<T> elementDtype)

Factory method to create a class to wrap a new TensorListConcat operation to the graph.

Parameters:

scope - current graph scope

inputHandle -

elementDtype -

Returns:

a new instance of TensorListConcat

tensor

public Output<T> tensor()

lengths

public Output<Long> lengths()