TPUPartitionedOutputV2

classe finale publique TPUPartitionedOutputV2

Une opération qui démultiplexe un tenseur à partitionner par XLA en une liste de partitionnés

sorties en dehors du calcul XLA. Prend en charge le partitionnement ND.

Méthodes publiques

statique <T> TPUPartitionedOutputV2 <T>
créer ( Scope scope, Operand <T> entrées, Long numSplits, List<Long> partitionDims)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPUPartitionedOutputV2.
Itérateur< Opérande <T>>
Liste< Sortie <T>>
sortir ()
Une liste de sorties partitionnées qui ont la même forme.

Méthodes héritées

Méthodes publiques

public static TPUPartitionedOutputV2 <T> create ( Scope scope, Operand <T> inputs, Long numSplits, List<Long> partitionDims)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération TPUPartitionedOutputV2.

Paramètres
portée portée actuelle
contributions Un tenseur qui représente la forme complète des tenseurs partitionnés.
partitionDims Une liste d'entiers décrivant comment chaque dimension est partitionnée. Le vide indique que les entrées sont répliquées.
Retour
  • une nouvelle instance de TPUPartitionedOutputV2

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

liste publique< Sortie <T>> sortie ()

Une liste de sorties partitionnées qui ont la même forme.