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