`reduction`を使用して` input`を `num_devices`から単一のデバイスに削減します。
`reduction`を使用して` input`を `num_devices`から単一のデバイスに削減します。
グラフは、すべての入力に有効なデバイス割り当てがあり、op自体にこれらのデバイスの1つが割り当てられるように作成する必要があります。
input:リダクションへの入力。データ:すべての `num_devices`デバイスにわたる削減の値。削減:実行する削減操作。
パブリックメソッド
出力<T> | asOutput () テンソルのシンボリックハンドルを返します。 |
static <T extends Number> NcclReduce <T> | |
出力<T> | データ() |
継承されたメソッド
パブリックメソッド
public Output <T> asOutput ()
テンソルのシンボリックハンドルを返します。
TensorFlow操作への入力は、別のTensorFlow操作の出力です。このメソッドは、入力の計算を表すシンボリックハンドルを取得するために使用されます。
public static NcclReduce <T> create (スコープスコープ、Iterable <オペランド<T >>入力、文字列削減)
新しいNcclReduce操作をラップするクラスを作成するファクトリメソッド。
パラメーター
範囲 | 現在のスコープ |
---|
戻り値
- NcclReduceの新しいインスタンス