XlaConcatND

public final class XlaConcatND

Concats input tensor across all dimensions.

An op which merges slices the input tensor based on the given num_splits attribute, strips paddings optionally, and returns the merged tensor without paddings.

This op may be generated via the TPU bridge.

For example, with `input` tensor:

[[0, 1],
  [4, 5]]
 [[2, 3],
  [6, 7]]
 [[8, 9],
  [12, 13]]
 [[10, 11],
  [14, 15]]
 

`num_splits`:

[2, 2]
 

and `paddings`:

[1, 1]
 

the expected `outputs` is:

[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]
 

Nested Classes

class

XlaConcatND.Options

Optional attributes for XlaConcatND

Public Methods

Output<T>	asOutput() Returns the symbolic handle of a tensor.
static <T> XlaConcatND<T>	create(Scope scope, Iterable<Operand<T>> inputs, List<Long> numConcats, Options... options) Factory method to create a class wrapping a new XlaConcatND operation.
Output<T>	output()
static XlaConcatND.Options	paddings(List<Long> paddings)

Inherited Methods

From class org.tensorflow.op.PrimitiveOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Returns the underlying Operation
final String	toString()

From class java.lang.Object

boolean	equals(Object arg0)
final Class<?>	getClass()
int	hashCode()
final void	notify()
final void	notifyAll()
String	toString()
final void	wait(long arg0, int arg1)
final void	wait(long arg0)
final void	wait()

From interface org.tensorflow.Operand

abstract Output<T>

asOutput() Returns the symbolic handle of a tensor.

Public Methods

public Output<T> asOutput ()

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.

public static XlaConcatND<T> create (Scope scope, Iterable<Operand<T>> inputs, List<Long> numConcats, Options... options)

Factory method to create a class wrapping a new XlaConcatND operation.

Parameters

scope	current scope
inputs	Input tensor slices in row-major order to merge across all dimensions. All inputs must have the same shape. } out_arg { name: "output" description: <
numConcats	Number of ways to merge per dimension.
options	carries optional attributes values

Returns

• a new instance of XlaConcatND

public Output<T> output ()

public static XlaConcatND.Options paddings (List<Long> paddings)

Parameters

paddings	Optional list of right paddings per dimension to strip from the final merged tensor. These paddings must not exceed the dimension size of the merged result prior to stripping paddings.