org.tensorflow.op.core

Class ParallelConcat<T>

java.lang.Object

org.tensorflow.op.PrimitiveOp

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

Type Parameters:

T - data type for output() output

All Implemented Interfaces:

Op, Operand<T>

@Operator
public final class ParallelConcat<T>
extends PrimitiveOp
implements Operand<T>

Concatenates a list of `N` tensors along the first dimension.

The input tensors are all required to have size 1 in the first dimension.

For example:

 # 'x' is [[1, 4]]
 # 'y' is [[2, 5]]
 # 'z' is [[3, 6]]
 parallel_concat([x, y, z]) => [[1, 4], [2, 5], [3, 6]]  # Pack along first dim.
 

The difference between concat and parallel_concat is that concat requires all of the inputs be computed before the operation will begin but doesn't require that the input shapes be known during graph construction. Parallel concat will copy pieces of the input into the output as they become available, in some situations this can provide a performance benefit.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
Output<T>	asOutput() Returns the symbolic handle of a tensor.
static <T> ParallelConcat<T>	create(Scope scope, Operand<T> values, Shape shape) Factory method to create a class to wrap a new ParallelConcat operation to the graph.
Output<T>	output() The concatenated 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> ParallelConcat<T> create(Scope scope,
                                           Operand<T> values,
                                           Shape shape)

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

Parameters:

scope - current graph scope

values - Tensors to be concatenated. All must have size 1 in the first dimension and same shape.

shape - the final shape of the result; should be equal to the shapes of any input but with the number of input values in the first dimension.

Returns:

a new instance of ParallelConcat

output

public Output<T> output()

The concatenated tensor.

asOutput

public Output<T> asOutput()

Description copied from interface: Operand

Returns the symbolic handle of a tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

Specified by:

asOutput in interface Operand<T>

See Also:

OperationBuilder.addInput(Output)