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

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
Gather.Options statique
batchDims (longs batchDims)
statique <T, U étend le nombre, V étend le nombre> Rassembler <T>
créer ( Portée de portée , paramètres d'opérande <T>, indices d'opérande <U>, axe d'opérande <V>, 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

Méthodes publiques

sortie publique <T> asOutput ()

Renvoie le handle symbolique d'un 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 portée , paramètres d'opérande <T>, indices d'opérande <U>, axe d'opérande <V>, 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:]`.