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