NcclReduce

公共finalクラスNcclReduce

`reduction`を使用して` input`を `num_devices`から単一のデバイスに削減します。

`reduction`を使用して` input`を `num_devices`から単一のデバイスに削減します。

グラフは、すべての入力に有効なデバイス割り当てがあり、op自体にこれらのデバイスの1つが割り当てられるように作成する必要があります。

input:リダクションへの入力。データ:すべての `num_devices`デバイスにわたる削減の値。削減:実行する削減操作。

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリックハンドルを返します。
静的<T拡張番号> NcclReduce <T>
作成スコープスコープ、反復処理可能<オペランド<T >>入力、文字列の還元)
新しいNcclReduce操作をラップするクラスを作成するファクトリメソッド。
出力<T>
データ()

継承されたメソッド

パブリックメソッド

公共の出力<T> asOutput()

テンソルのシンボリックハンドルを返します。

TensorFlow操作への入力は、別のTensorFlow操作の出力です。このメソッドは、入力の計算を表すシンボリックハンドルを取得するために使用されます。

パブリック静的NcclReduce <T>を作成スコープスコープ、反復処理可能<オペランド<T >>入力、文字列の還元)

新しいNcclReduce操作をラップするクラスを作成するファクトリメソッド。

パラメーター
範囲現在のスコープ
戻り値
  • NcclReduceの新しいインスタンス

公共の出力<T>データ()