Cette page a été traduite par l'API Cloud Translation.
Switch to English

tensorflow :: ops :: GatherV2

#include <array_ops.h>

Rassembler les tranches de params de axis selon les indices .

Résumé

indices doit être un tenseur de nombre entier de la plus grande dimension (généralement 0-D-1 ou 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 le CPU, si un indice de limite se trouve, une erreur est renvoyée. Le GPU, si un index de limite se trouve, un 0 est stockée dans la valeur de sortie correspondante.

Voir aussi tf.batch_gather et tf.gather_nd .

Arguments:

  • portée: A Portée objet
  • params: Le tenseur à partir desquels les valeurs. Doit être au moins rang axis + 1 .
  • indices: Indice tenseur. Doit être à portée [0, params.shape[axis]) .
  • axe: L'axe de params pour recueillir des indices de. Par défaut, la première dimension. Prise en charge des index négatifs.

Retour:

  • Output : Les valeurs de params recueillies à partir des indices donnés par des indices , avec la forme params.shape[:axis] + indices.shape + params.shape[axis + 1:] .

Constructeurs et Destructeurs

GatherV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, :: tensorflow::Input axis)
GatherV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, :: tensorflow::Input axis, const GatherV2::Attrs & attrs)

attributs publics

operation
output

Les fonctions publiques

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

fonctions statiques publiques

BatchDims (int64 x)

struct

tensorflow :: :: ops GatherV2 :: ATTRS

Setters d'attributs en option pour GatherV2 .

attributs publics

opération

 Operation operation

production

:: tensorflow::Output output

Les fonctions publiques

GatherV2

 GatherV2(
  const :: tensorflow::Scope & scope,
  :: tensorflow::Input params,
  :: tensorflow::Input indices,
  :: tensorflow::Input axis
)

GatherV2

 GatherV2(
  const :: tensorflow::Scope & scope,
  :: tensorflow::Input params,
  :: tensorflow::Input indices,
  :: tensorflow::Input axis,
  const GatherV2::Attrs & attrs
)

nœud

::tensorflow::Node * node() const 

opérateur :: tensorflow :: entrée

 operator::tensorflow::Input() const 

opérateur :: tensorflow :: sortie

 operator::tensorflow::Output() const 

fonctions statiques publiques

BatchDims

 Attrs BatchDims(
  int64 x
)