DynamicPartition

classe finale publique DynamicPartition

Partitionne les « données » en tenseurs « num_partitions » en utilisant les indices des « partitions ».

Pour chaque tuple d'index `js` de taille `partitions.ndim`, la tranche `data[js, ...]` fait partie de `outputs[partitions[js]]`. Les tranches avec `partitions[js] = i` sont placées dans `outputs[i]` dans l'ordre lexicographique de `js`, et la première dimension de `outputs[i]` est le nombre d'entrées dans `partitions` égal à 'je'. En détail,

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

`data.shape` doit commencer par `partitions.shape`.

Par exemple :

# 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]
 

Voir `dynamic_stitch` pour un exemple sur la façon de fusionner des partitions.

Méthodes publiques

statique <T> DynamicPartition <T>	créer (portée de portée , données d'opérande <T>, partitions d'opérande <Integer>, numPartitions longues) Méthode Factory pour créer une classe encapsulant une nouvelle opération DynamicPartition.
Itérateur< Opérande <T>>	itérateur ()
Liste< Sortie <T>>	les sorties ()

Méthodes héritées

De la classe org.tensorflow.op.PrimitiveOp

booléen final	est égal (Obj objet)
entier final	Code de hachage ()
Opération	op () Renvoie l' Operation sous-jacente
Chaîne finale	àChaîne ()

De la classe java.lang.Object

booléen	est égal à (Objet arg0)
Classe finale<?>	obtenirClasse ()
int	Code de hachage ()
vide final	notifier ()
vide final	notifierTous ()
Chaîne	àChaîne ()
vide final	attendre (long arg0, int arg1)
vide final	attendez (long arg0)
vide final	attendez ()

Depuis l'interface java.lang.Iterable

vide abstrait	forEach (Consommateur <? super T> arg0)
Itérateur abstrait < Opérande <T>>	itérateur ()
Spliterator abstrait < Opérande <T>>	séparateur ()

Méthodes publiques

public static DynamicPartition <T> créer (portée de portée , données d'opérande <T>, partitions d'opérande <Integer>, numPartitions longues)

Méthode Factory pour créer une classe encapsulant une nouvelle opération DynamicPartition.

Paramètres

portée	portée actuelle
cloisons	N'importe quelle forme. Indices dans la plage `[0, num_partitions)`.
numPartitions	Le nombre de partitions à générer.

Retour

• une nouvelle instance de DynamicPartition

public Iterator< Opérande <T>> itérateur ()

liste publique < Sortie <T>> sorties ()