org.bytedeco.javacpp

Class tensorflow.GatherNd

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.javacpp.tensorflow.GatherNd

All Implemented Interfaces:

AutoCloseable

Enclosing class:

tensorflow

@Namespace(value="tensorflow::ops")
 @NoOffset
public static class tensorflow.GatherNd
extends Pointer

Gather slices from params into a Tensor with shape specified by indices. indices is an K-dimensional integer tensor, best thought of as a (K-1)-dimensional tensor of indices into params, where each element defines a slice of params: output[\\(i_0, ..., i_{K-2}\\)] = params[indices[\\(i_0, ..., i_{K-2}\\)]] Whereas in tf.gather indices defines slices into the first dimension of params, in tf.gather_nd, indices defines slices into the first N dimensions of params, where N = indices.shape[-1]. The last dimension of indices can be at most the rank of params: indices.shape[-1] <= params.rank The last dimension of indices corresponds to elements (if indices.shape[-1] == params.rank) or slices (if indices.shape[-1] < params.rank) along dimension indices.shape[-1] of params. The output tensor has shape indices.shape[:-1] + params.shape[indices.shape[-1]:] Note that on CPU, if an out of bound index is found, an error is returned. On GPU, if an out of bound index is found, a 0 is stored in the corresponding output value. Some examples below. Simple indexing into a matrix:

 python
      indices = [[0, 0], [1, 1]]
      params = [['a', 'b'], ['c', 'd']]
      output = ['a', 'd']
  

Slice indexing into a matrix:

 python
      indices = [[1], [0]]
      params = [['a', 'b'], ['c', 'd']]
      output = [['c', 'd'], ['a', 'b']]
  

Indexing into a 3-tensor:

 python
      indices = [[1]]
      params = [[['a0', 'b0'], ['c0', 'd0']],
                [['a1', 'b1'], ['c1', 'd1']]]
      output = [[['a1', 'b1'], ['c1', 'd1']]]
 
 
      indices = [[0, 1], [1, 0]]
      params = [[['a0', 'b0'], ['c0', 'd0']],
                [['a1', 'b1'], ['c1', 'd1']]]
      output = [['c0', 'd0'], ['a1', 'b1']]
 
 
      indices = [[0, 0, 1], [1, 0, 1]]
      params = [[['a0', 'b0'], ['c0', 'd0']],
                [['a1', 'b1'], ['c1', 'd1']]]
      output = ['b0', 'b1']
  

Batched indexing into a matrix:

 python
      indices = [[[0, 0]], [[0, 1]]]
      params = [['a', 'b'], ['c', 'd']]
      output = [['a'], ['b']]
  

Batched slice indexing into a matrix:

 python
      indices = [[[1]], [[0]]]
      params = [['a', 'b'], ['c', 'd']]
      output = [[['c', 'd']], [['a', 'b']]]
  

Batched indexing into a 3-tensor:

 python
      indices = [[[1]], [[0]]]
      params = [[['a0', 'b0'], ['c0', 'd0']],
                [['a1', 'b1'], ['c1', 'd1']]]
      output = [[[['a1', 'b1'], ['c1', 'd1']]],
                [[['a0', 'b0'], ['c0', 'd0']]]]
 
      indices = [[[0, 1], [1, 0]], [[0, 0], [1, 1]]]
      params = [[['a0', 'b0'], ['c0', 'd0']],
                [['a1', 'b1'], ['c1', 'd1']]]
      output = [[['c0', 'd0'], ['a1', 'b1']],
                [['a0', 'b0'], ['c1', 'd1']]]
 
 
      indices = [[[0, 0, 1], [1, 0, 1]], [[0, 1, 1], [1, 1, 0]]]
      params = [[['a0', 'b0'], ['c0', 'd0']],
                [['a1', 'b1'], ['c1', 'd1']]]
      output = [['b0', 'b1'], ['d0', 'c1']]
  

See also tf.gather and tf.batch_gather. Arguments: * scope: A Scope object * params: The tensor from which to gather values. * indices: Index tensor. Returns: * Output: Values from params gathered from indices given by indices, with shape indices.shape[:-1] + params.shape[indices.shape[-1]:].

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
GatherNd(Pointer p) Pointer cast constructor.
GatherNd(tensorflow.Scope scope, tensorflow.Input params, tensorflow.Input indices)

Method Summary

Modifier and Type	Method and Description
tensorflow.Input	asInput()
tensorflow.Output	asOutput()
tensorflow.Node	node()
tensorflow.Operation	operation()
tensorflow.GatherNd	operation(tensorflow.Operation operation)
tensorflow.Output	output()
tensorflow.GatherNd	output(tensorflow.Output output)

Methods inherited from class org.bytedeco.javacpp.Pointer

address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, hashCode, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetof, parseBytes, physicalBytes, position, position, put, realloc, setNull, sizeof, toString, totalBytes, totalPhysicalBytes, withDeallocator, zero

Methods inherited from class java.lang.Object

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

Constructor Detail

GatherNd

public GatherNd(Pointer p)

Pointer cast constructor. Invokes Pointer.Pointer(Pointer).

GatherNd

public GatherNd(@Const @ByRef
                tensorflow.Scope scope,
                @ByVal
                tensorflow.Input params,
                @ByVal
                tensorflow.Input indices)

Method Detail

asOutput

@ByVal
 @Name(value="operator tensorflow::Output")
public tensorflow.Output asOutput()

asInput

@ByVal
 @Name(value="operator tensorflow::Input")
public tensorflow.Input asInput()

node

public tensorflow.Node node()

operation

@ByRef
public tensorflow.Operation operation()

operation

public tensorflow.GatherNd operation(tensorflow.Operation operation)

output

@ByRef
public tensorflow.Output output()

output

public tensorflow.GatherNd output(tensorflow.Output output)