AssignVariableXlaConcatND

public final class AssignVariableXlaConcatND

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 writes the merged tensor without paddings to the resource variable.

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 AssignVariableXlaConcatND.Options Optional attributes for AssignVariableXlaConcatND  

Public Methods

static <T> AssignVariableXlaConcatND
create(Scope scope, Operand<?> resource, Iterable<Operand<T>> inputs, List<Long> numConcats, Options... options)
Factory method to create a class wrapping a new AssignVariableXlaConcatND operation.
static AssignVariableXlaConcatND.Options
paddings(List<Long> paddings)

Inherited Methods

org.tensorflow.op.PrimitiveOp
final boolean
equals(Object obj)
final int
hashCode()
Operation
op()
Returns the underlying Operation
final String
toString()
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()

Public Methods

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

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

Parameters
scope current scope
resource Resource variable for concatenated input tensors across all dimensions. } in_arg { name: "inputs" description: <
numConcats Number of ways to merge per dimension.
options carries optional attributes values
Returns
  • a new instance of AssignVariableXlaConcatND

public static AssignVariableXlaConcatND.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.