tensorflow::ops::BarrierTakeMany

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

Résumé

Cette opération concatène les tenseurs de composants d'éléments complets le long de la dimension 0 pour former un tenseur de composant unique.

Les éléments sortent de la barrière une fois complets, et dans l'ordre où ils y ont été insérés. Les indices de sortie fournissent des informations sur le lot auquel chaque élément a été initialement inséré dans la barrière.

Arguments :

• portée : 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 : Autoriser le retour d’un nombre d’éléments inférieur à num_elements si la barrière est déjà fermée.
• timeout_ms : Si la file d’attente est vide, cette opération sera bloquée pendant un délai maximal de timeout_ms millisecondes. Remarque : Cette option n’est pas encore prise en charge.

Retours :

• Indices 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 (en commençant par MIN_LONG et en augmentant avec chaque BarrierInsertMany ).
• Clés Output : un tenseur unidimensionnel de clés, de longueur num_elements.
• Valeurs OutputList : un tenseur de dimension quelconque par composante d’un élément de barrière. Toutes les valeurs ont une longueur de num_elements dans la dimension 0.

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
)