パブリック最終クラスNcclReduce
「reduction」を使用して「num_devices」から「input」を単一のデバイスに減らします。
「reduction」を使用して「num_devices」から「input」を単一のデバイスに減らします。
グラフは、すべての入力に有効なデバイス割り当てがあり、演算自体がこれらのデバイスのいずれかに割り当てられるように構築する必要があります。
input: リダクションへの入力。 data: すべての「num_devices」デバイスにわたる削減の値。 duction: 実行するリダクション操作。
定数
弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
出力<T> | asOutput () テンソルのシンボリック ハンドルを返します。 |
static <T extends TNumber > NcclReduce <T> | |
出力<T> | データ() |
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
定数値: "NcclReduce"
パブリックメソッド
public Output <T> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static NcclReduce <T> create (スコープscope、Iterable< Operand <T>>入力、文字列リダクション)
新しい NcclReduce オペレーションをラップするクラスを作成するためのファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|
戻り値
- NcclReduce の新しいインスタンス