テンソルフロー::作戦::バリアテイクメニー

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

まとめ

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

要素は、完成するとバリア内に配置された順序でバリアから出てきます。インデックス出力は、各要素が最初にバリアに挿入されたバッチに関する情報を提供します。

引数:

• スコープ:スコープオブジェクト
• handle: バリアのハンドル。
• 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
)