Gather

cours final public

Encapsule l'opérateur XLA Gather documenté dans

https://www.tensorflow.org/xla/operation_semantics#gather

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique du tenseur.
static <T étend TType , U étend TNumber > Gather <T>
créer ( Scope scope, Operand <T> operand, Operand <U> startIndices, Operand <U> sliceSizes, String dimensionNumbers, Boolean indicesAreSorted)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération Gather.
Sortie <T>
sortir ()

Méthodes héritées

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : "XlaGather"

Méthodes publiques

sortie publique <T> asOutput ()

Renvoie le handle symbolique du tenseur.

Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.

public static Gather <T> create ( Scope scope, Operand <T> operand, Operand <U> startIndices, Operand <U> sliceSizes, String dimensionNumbers, Boolean indicesAreSorted)

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

Paramètres
portée portée actuelle
opérande Le tableau à partir duquel nous rassemblons.
démarrerIndices Tableau contenant les indices de départ des tranches que nous collectons.
Tailles de tranches slice_sizes[i] est les limites de la tranche sur la dimension i.
DimensionsNuméros Un proto xla::GatherDimensionNumbers sérialisé.
indicesAreSorted Booléen indiquant si les index sont triés.
Retour
  • une nouvelle instance de Gather

sortie publique <T> sortie ()