flux tenseur :: opérations :: BarrièrePrendreBeaucoup

Prend le nombre donné d'éléments terminés d'une barrière.

Sommaire

Cette opération concatène les tenseurs de composants d'éléments complétés le long de la 0ème dimension pour créer un seul tenseur de composants.

Les éléments sortent de la barrière lorsqu'ils sont complets et dans l'ordre dans lequel ils ont été placés dans la barrière. La sortie des indices fournit des informations sur le lot dans lequel chaque élément a été inséré à l'origine dans la barrière.

Arguments:

• scope : un objet Scope
• poignée : la poignée d'une barrière.
• num_elements : un tenseur à un seul élément contenant le nombre d'éléments à prendre.
• component_types : le type de chaque composant dans une valeur.

Attributs optionnels (voir Attrs ) :

• allow_small_batch : permet de renvoyer moins de num_elements éléments si la barrière est déjà fermée.
• timeout_ms : si la file d'attente est vide, cette opération bloquera jusqu'à timeout_ms millisecondes. Remarque : Cette option n'est pas encore prise en charge.

Retour:

• Indices de Output : un tenseur unidimensionnel d'indices, de longueur num_elems. Ces indices font référence au lot dans lequel les valeurs ont été placées dans la barrière (commençant par MIN_LONG et augmentant avec chaque BarrierInsertMany ).
• Clés de Output : un tenseur unidimensionnel de clés, de longueur num_elements.
• Valeurs OutputList : un tenseur de toute dimension par composant dans un élément barrière. Toutes les valeurs ont une longueur de num_elements dans la 0e dimension.

Attributs publics

::tensorflow::Output indices

::tensorflow::Output keys

Operation operation

::tensorflow::OutputList values

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types
)

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types,
  const BarrierTakeMany::Attrs & attrs
)

Attrs AllowSmallBatch(
  bool x
)

Attrs TimeoutMs(
  int64 x
)

Attrs WaitForIncomplete(
  bool x
)