tensorflow:: ops:: BarrierTakeMany

バリアから指定された数の完成済み要素を取得します。

まとめ

この操作は、0次元に沿って要素が完成したコンポーネントテンソルを連結し、単一のコンポーネントテンソルを作成します。

要素は、処理が完了すると、バリアに挿入された順序でバリアから取り出されます。出力されるインデックスには、各要素が最初にバリアに挿入されたバッチに関する情報が含まれています。

議論:

• スコープ:スコープオブジェクト
• 取っ手：障壁の取っ手。
• num_elements: 取得する要素数を含む単一要素テンソル。
• component_types: 値に含まれる各コンポーネントのタイプ。

オプション属性（ Attrs参照）：

• allow_small_batch: バリアが既に閉じている場合、num_elements 個未満のアイテムを返すことを許可します。
• timeout_ms: キューが空の場合、この操作は最大 timeout_ms ミリ秒間ブロックされます。注: このオプションはまだサポートされていません。

返品:

• Outputインデックス: 長さ num_elems のインデックスの 1 次元テンソル。これらのインデックスは、値がバリアに配置されたバッチ (MIN_LONG から始まり、 BarrierInsertManyごとに増加) を参照します。
• Outputキー：長さ num_elements のキーの 1 次元テンソル。
• OutputList値：バリア要素内の各コンポーネントにつき、任意の次元のテンソルが1つ含まれます。すべての値は、0次元目において長さがnum_elementsとなります。

公開属性

::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
)