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

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

Résumé

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

Les éléments sortent de la barrière lorsqu'ils sont terminés 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é initialement inséré dans la barrière.

Arguments :

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

Attributs facultatifs (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.

Retours :

• Indices Output : un tenseur d'indices unidimensionnel, de longueur num_elems. Ces indices font référence au lot dans lequel les valeurs ont été placées dans la barrière (en commençant par MIN_LONG et en augmentant à chaque BarrierInsertMany ).
• Clés Output : un tenseur de clés unidimensionnel, 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 num_elements dans la 0ème 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
)