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

#include <data_flow_ops.h>

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

まとめ

この操作は、完成した要素のコンポーネント テンソルを 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 です。

コンストラクターとデストラクター

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)

パブリック属性

indices
keys
operation
values

パブリック静的関数

AllowSmallBatch (bool x)
TimeoutMs (int64 x)
WaitForIncomplete (bool x)

構造体

tensorflow:: ops:: BarrierTakeMany:: Attrs

BarrierTakeManyのオプションの属性セッター。

パブリック属性

インデックス

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

パブリック静的関数

許可SmallBatch

Attrs AllowSmallBatch(
  bool x
)

タイムアウトさん

Attrs TimeoutMs(
  int64 x
)

未完了を待つ

Attrs WaitForIncomplete(
  bool x
)