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