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
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> 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:]`.