org.bytedeco.javacpp

Class tensorflow.DynamicPartition

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.javacpp.tensorflow.DynamicPartition

All Implemented Interfaces:

AutoCloseable

Enclosing class:

tensorflow

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

Partitions data into num_partitions tensors using indices from partitions. For each index tuple js of size partitions.ndim, the slice data[js, ...] becomes part of outputs[partitions[js]]. The slices with partitions[js] = i are placed in outputs[i] in lexicographic order of js, and the first dimension of outputs[i] is the number of entries in partitions equal to i. In detail,

 python
      outputs[i].shape = [sum(partitions == i)] + data.shape[partitions.ndim:]
 
      outputs[i] = pack([data[js, ...] for js if partitions[js] == i])
  

data.shape must start with partitions.shape. For example:

 python
      # Scalar partitions.
      partitions = 1
      num_partitions = 2
      data = [10, 20]
      outputs[0] = []  # Empty with shape [0, 2]
      outputs[1] = [[10, 20]]
 
      # Vector partitions.
      partitions = [0, 0, 1, 1, 0]
      num_partitions = 2
      data = [10, 20, 30, 40, 50]
      outputs[0] = [10, 20, 50]
      outputs[1] = [30, 40]
  

See dynamic_stitch for an example on how to merge partitions back.

Arguments: * scope: A Scope object * partitions: Any shape. Indices in the range [0, num_partitions). * num_partitions: The number of partitions to output. Returns: * OutputList: The outputs tensor.

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
DynamicPartition(Pointer p) Pointer cast constructor.
DynamicPartition(tensorflow.Scope scope, tensorflow.Input data, tensorflow.Input partitions, long num_partitions)

Method Summary

Modifier and Type	Method and Description
tensorflow.Output	get(long index)
tensorflow.Operation	operation()
tensorflow.DynamicPartition	operation(tensorflow.Operation operation)
tensorflow.OutputVector	outputs()
tensorflow.DynamicPartition	outputs(tensorflow.OutputVector outputs)

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

DynamicPartition

public DynamicPartition(Pointer p)

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

DynamicPartition

public DynamicPartition(@Const @ByRef
                        tensorflow.Scope scope,
                        @ByVal
                        tensorflow.Input data,
                        @ByVal
                        tensorflow.Input partitions,
                        @Cast(value="tensorflow::int64")
                        long num_partitions)

Method Detail

get

@ByVal
 @Name(value="operator []")
public tensorflow.Output get(@Cast(value="size_t")
                                                                  long index)

operation

@ByRef
public tensorflow.Operation operation()

operation

public tensorflow.DynamicPartition operation(tensorflow.Operation operation)

outputs

@ByRef
 @Cast(value="tensorflow::OutputList*")
public tensorflow.OutputVector outputs()

outputs

public tensorflow.DynamicPartition outputs(tensorflow.OutputVector outputs)