NcclBroadcast

パブリック最終クラスNcclBroadcast

出力に接続されているすべてのデバイスに「入力」を送信します。

出力に接続されているすべてのデバイスに「入力」を送信します。

グラフは、出力に接続されているすべての演算に有効なデバイス割り当てがあり、演算自体がこれらのデバイスのいずれかに割り当てられるように構築する必要があります。

input: ブロードキャストへの入力。出力: 入力と同じ。形状: 入力テンソルの形状。

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリック ハンドルを返します。
static <T extends Number> NcclBroadcast <T>
create ( Scopeスコープ、オペランド<T> 入力、 Shapeシェイプ)
新しい NcclBroadcast オペレーションをラップするクラスを作成するためのファクトリ メソッド。
出力<T>

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

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

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

public static NcclBroadcast <T> create ( Scopeスコープ、オペランド<T> 入力、 Shapeシェイプ)

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

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

public出力<T>出力()