CollectiveGather

パブリック最終クラスCollectiveGather

同じタイプと形状の複数のテンソルを相互に蓄積します。

ネストされたクラス

クラスCollectiveGather.オプションCollectiveGatherのオプションの属性

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリック ハンドルを返します。
静的CollectiveGather.Options
communicationHint (文字列 communicationHint)
static <T extends Number> CollectiveGather <T>
create ( Scopeスコープ、オペランド<T> 入力、Long groupSize、Long groupKey、Long instanceKey、 Shapeシェイプ、 Options...オプション)
新しい CollectiveGather オペレーションをラップするクラスを作成するためのファクトリ メソッド。
出力<T>
データ()
静的CollectiveGather.Options
timeoutSeconds (浮動小数点タイムアウト秒)

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

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

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

public static CollectiveGather.Options communicationHint (String communicationHint)

public static CollectiveGather <T> create (スコープスコープ、オペランド<T> 入力、Long groupSize、Long groupKey、Long instanceKey、 Shapeシェイプ、 Options...オプション)

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

パラメーター
範囲現在のスコープ
オプションオプションの属性値を持ちます
戻り値
  • CollectiveGather の新しいインスタンス

public出力<T>データ()

public static CollectiveGather.Options timeoutSeconds (Float timeoutSeconds)