Gather

cours final public

Rassemblez les tranches de l'axe « params » selon les « indices ».

Les « indices » doivent être un tenseur entier de n'importe quelle dimension (généralement 0-D ou 1-D). Produit un tenseur de sortie avec la forme `params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]` où :

# Scalar indices (output is rank(params) - 1).
     output[a_0, ..., a_n, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices, b_0, ..., b_n]
 
     # Vector indices (output is rank(params)).
     output[a_0, ..., a_n, i, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
 
     # Higher rank indices (output is rank(params) + rank(indices) - 1).
     output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
       params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]
 

Notez que sur le CPU, si un index hors limite est trouvé, une erreur est renvoyée. Sur GPU, si un index hors limite est trouvé, un 0 est stocké dans la valeur de sortie correspondante.

Voir aussi `tf.batch_gather` et tf.gather_nd .

Classes imbriquées

classe Rassembler.Options Attributs facultatifs pour 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.
Gather.Options statique
batchDims (longs batchDims)
statique <T étend TType > Rassembler <T>
créer ( Portée de la portée , Paramètres de l'opérande <T>, Opérande <? étend TNumber > indices, Opérande <? étend TNumber > axe, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération Gather.
Sortie <T>
sortir ()
Valeurs de `params` recueillies à partir des indices donnés par `indices`, avec la forme `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`.

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 : "GatherV2"

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 statique Gather.Options batchDims (Long batchDims)

public static Gather <T> créer (portée de la portée , paramètres de l'opérande <T>, opérande <? étend TNumber > indices, opérande <? étend TNumber > axe, options... options)

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

Paramètres
portée portée actuelle
paramètres Tenseur à partir duquel collecter les valeurs. Doit être au moins de rang « axe + 1 ».
indices Tenseur d'indice. Doit être dans la plage `[0, params.shape[axis])`.
axe L'axe dans `params` à partir duquel collecter les `indices`. La valeur par défaut est la première dimension. Prend en charge les index négatifs.
choix porte des valeurs d'attributs facultatifs
Retour
  • une nouvelle instance de Gather

sortie publique <T> sortie ()

Valeurs de `params` recueillies à partir des indices donnés par `indices`, avec la forme `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`.