NcclAllReduce

パブリック最終クラスNcclAllReduce

すべての入力テンソルにわたるリダクションを含むテンソルを出力します。

同じ `shared_name 内の ops に渡されるすべての入力テンソルにわたるリダクションを含むテンソルを出力します。

グラフは、1 つの操作がshared_name値「c」で実行される場合、「num_devices」操作がshared_name値「c」で実行されるように構築する必要があります。そうしないと、グラフの実行が完了できなくなります。

input: リダクションへの入力 data: すべての `num_devices` デバイスにわたるリダクションの値。 duction: 実行するリダクション操作。 num_devices: この削減に参加するデバイスの数。 shared_name: 同じリダクションの操作間で共有される識別子。

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリック ハンドルを返します。
static <T extends Number> NcclAllReduce <T>
create (スコープscope、オペランド<T>入力、文字列リダクション、Long numDevices、StringsharedName)
新しい NcclAllReduce オペレーションをラップするクラスを作成するためのファクトリ メソッド。
出力<T>
データ()

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

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

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

public static NcclAllReduce <T> create (スコープscope、オペランド<T>入力、文字列リダクション、Long numDevices、StringsharedName)

新しい NcclAllReduce オペレーションをラップするクラスを作成するためのファクトリ メソッド。

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

public出力<T>データ()