パブリック最終クラスBarrierTakeMany
バリアから指定された数の完了した要素を取得します。
この操作は、完成した要素のコンポーネント テンソルを 0 次元に沿って連結して、単一のコンポーネント テンソルを作成します。
要素は、完成するとバリア内に配置された順序でバリアから出てきます。インデックス出力は、各要素が最初にバリアに挿入されたバッチに関する情報を提供します。
ネストされたクラス
クラス | BarrierTakeMany.オプション | BarrierTakeMany のオプションの属性 |
定数
弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
静的BarrierTakeMany.Options | allowSmallBatch (ブール値のallowSmallBatch) |
静的BarrierTakeMany | |
出力< TInt64 > | インデックス() 長さ num_elems のインデックスの 1 次元テンソル。 |
出力< TString > | キー() 長さ num_elements のキーの 1 次元テンソル。 |
静的BarrierTakeMany.Options | timeoutMs (ロングタイムアウトMs) |
リスト<出力<?>> | 値() バリア要素のコンポーネントごとに 1 つの任意次元テンソル。 |
静的BarrierTakeMany.Options | waitForIncomplete (ブール値 waitForIncomplete) |
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
定数値: "BarrierTakeMany"
パブリックメソッド
public static BarrierTakeMany.OptionsallowSmallBatch (ブール値のallowSmallBatch)
パラメーター
allowSmallBatch | バリアがすでに閉じられている場合、num_elements 未満の項目を返すことを許可します。 |
---|
public static BarrierTakeMany create (スコープスコープ、オペランド< TString > ハンドル、オペランド< TInt32 > numElements、List<Class<? extends TType >>ComponentTypes、 Options...オプション)
新しい BarrierTakeMany オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
ハンドル | バリアのハンドル。 |
要素数 | 取得する要素の数を含む単一要素テンソル。 |
コンポーネントの種類 | 値の各コンポーネントのタイプ。 |
オプション | オプションの属性値を持ちます |
戻り値
- BarrierTakeMany の新しいインスタンス
public出力< TInt64 >インデックス()
長さ num_elems のインデックスの 1 次元テンソル。これらのインデックスは、値がバリアに配置されたバッチを参照します (MIN_LONG から始まり、BarrierInsertMany ごとに増加します)。
public static BarrierTakeMany.Options timeoutMs (長いタイムアウトMs)
パラメーター
タイムアウトさん | キューが空の場合、この操作は最大 timeout_ms ミリ秒間ブロックされます。注: このオプションはまだサポートされていません。 |
---|