DynamicPartition
Stay organized with collections
Save and categorize content based on your preferences.
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,
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:
# 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.
Constants
String |
OP_NAME |
The name of this op, as known by TensorFlow core engine |
Inherited Methods
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
java.lang.Iterable
abstract
void
|
forEach(Consumer<? super T> arg0)
|
abstract
Iterator<Operand<T>>
|
iterator()
|
abstract
Spliterator<Operand<T>>
|
spliterator()
|
Constants
public
static
final
String
OP_NAME
The name of this op, as known by TensorFlow core engine
Constant Value:
"DynamicPartition"
Public Methods
Factory method to create a class wrapping a new DynamicPartition operation.
Parameters
scope |
current scope |
partitions |
Any shape. Indices in the range `[0, num_partitions)`. |
numPartitions |
The number of partitions to output. |
Returns
- a new instance of DynamicPartition
public
Iterator<Operand<T>>
iterator
()
public
List<Output<T>>
outputs
()
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2021-11-29 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2021-11-29 UTC."],[],[],null,["# DynamicPartition\n\npublic final class **DynamicPartition** \nPartitions \\`data\\` into \\`num_partitions\\` tensors using indices from \\`partitions\\`.\n\n\nFor each index tuple \\`js\\` of size \\`partitions.ndim\\`, the slice \\`data\\[js, ...\\]\\`\nbecomes part of \\`outputs\\[partitions\\[js\\]\\]\\`. The slices with \\`partitions\\[js\\] = i\\`\nare placed in \\`outputs\\[i\\]\\` in lexicographic order of \\`js\\`, and the first\ndimension of \\`outputs\\[i\\]\\` is the number of entries in \\`partitions\\` equal to \\`i\\`.\nIn detail, \n\n outputs[i].shape = [sum(partitions == i)] + data.shape[partitions.ndim:]\n \n outputs[i] = pack([data[js, ...] for js if partitions[js] == i])\n \n\\`data.shape\\` must start with \\`partitions.shape\\`.\n\n\nFor example: \n\n # Scalar partitions.\n partitions = 1\n num_partitions = 2\n data = [10, 20]\n outputs[0] = [] # Empty with shape [0, 2]\n outputs[1] = [[10, 20]]\n \n # Vector partitions.\n partitions = [0, 0, 1, 1, 0]\n num_partitions = 2\n data = [10, 20, 30, 40, 50]\n outputs[0] = [10, 20, 50]\n outputs[1] = [30, 40]\n \nSee \\`dynamic_stitch\\` for an example on how to merge partitions back.\n\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n### Constants\n\n|--------|-------------------------------------------------------------------------------|---------------------------------------------------------|\n| String | [OP_NAME](/jvm/api_docs/java/org/tensorflow/op/core/DynamicPartition#OP_NAME) | The name of this op, as known by TensorFlow core engine |\n\n### Public Methods\n\n|------------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| static \\\u003cT extends [TType](/jvm/api_docs/java/org/tensorflow/types/family/TType)\\\u003e [DynamicPartition](/jvm/api_docs/java/org/tensorflow/op/core/DynamicPartition)\\\u003cT\\\u003e | [create](/jvm/api_docs/java/org/tensorflow/op/core/DynamicPartition#create(org.tensorflow.op.Scope, org.tensorflow.Operand\u003cT\u003e, org.tensorflow.Operand\u003corg.tensorflow.types.TInt32\u003e, java.lang.Long))([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003cT\\\u003e data, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TInt32](/jvm/api_docs/java/org/tensorflow/types/TInt32)\\\u003e partitions, Long numPartitions) Factory method to create a class wrapping a new DynamicPartition operation. |\n| Iterator\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003cT\\\u003e\\\u003e | [iterator](/jvm/api_docs/java/org/tensorflow/op/core/DynamicPartition#iterator())() |\n| List\\\u003c[Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e\\\u003e | [outputs](/jvm/api_docs/java/org/tensorflow/op/core/DynamicPartition#outputs())() |\n\n### Inherited Methods\n\nFrom class [org.tensorflow.op.RawOp](/jvm/api_docs/java/org/tensorflow/op/RawOp) \n\n|----------------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| final boolean | [equals](/jvm/api_docs/java/org/tensorflow/op/RawOp#equals(java.lang.Object))(Object obj) |\n| final int | [hashCode](/jvm/api_docs/java/org/tensorflow/op/RawOp#hashCode())() |\n| [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/RawOp#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n| final String | [toString](/jvm/api_docs/java/org/tensorflow/op/RawOp#toString())() |\n\nFrom class java.lang.Object \n\n|------------------|---------------------------|\n| boolean | equals(Object arg0) |\n| final Class\\\u003c?\\\u003e | getClass() |\n| int | hashCode() |\n| final void | notify() |\n| final void | notifyAll() |\n| String | toString() |\n| final void | wait(long arg0, int arg1) |\n| final void | wait(long arg0) |\n| final void | wait() |\n\nFrom interface [org.tensorflow.op.Op](/jvm/api_docs/java/org/tensorflow/op/Op) \n\n|-----------------------------------------------------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| abstract [ExecutionEnvironment](/jvm/api_docs/java/org/tensorflow/ExecutionEnvironment) | [env](/jvm/api_docs/java/org/tensorflow/op/Op#env())() Return the execution environment this op was created in. |\n| abstract [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/Op#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n\nFrom interface java.lang.Iterable \n\n|-----------------------------------------------------------------------------------|-------------------------------------|\n| abstract void | forEach(Consumer\\\u003c? super T\\\u003e arg0) |\n| abstract Iterator\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003cT\\\u003e\\\u003e | iterator() |\n| abstract Spliterator\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003cT\\\u003e\\\u003e | spliterator() |\n\nConstants\n---------\n\n#### public static final String\n**OP_NAME**\n\nThe name of this op, as known by TensorFlow core engine \nConstant Value: \"DynamicPartition\"\n\nPublic Methods\n--------------\n\n#### public static [DynamicPartition](/jvm/api_docs/java/org/tensorflow/op/core/DynamicPartition)\\\u003cT\\\u003e\n**create**\n([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003cT\\\u003e data, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TInt32](/jvm/api_docs/java/org/tensorflow/types/TInt32)\\\u003e partitions, Long numPartitions)\n\nFactory method to create a class wrapping a new DynamicPartition operation. \n\n##### Parameters\n\n| scope | current scope |\n| partitions | Any shape. Indices in the range \\`\\[0, num_partitions)\\`. |\n| numPartitions | The number of partitions to output. |\n|---------------|-----------------------------------------------------------|\n\n##### Returns\n\n- a new instance of DynamicPartition \n\n#### public Iterator\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003cT\\\u003e\\\u003e\n**iterator**\n()\n\n\u003cbr /\u003e\n\n#### public List\\\u003c[Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e\\\u003e\n**outputs**\n()\n\n\u003cbr /\u003e"]]