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> | |
Itérateur< Opérande <T>> | itérateur () |
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