CollectiveGather

パブリックファイナルクラスCollectiveGather

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

ネストされたクラス

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

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリックハンドルを返します。
static CollectiveGather.Options
communicationHint (文字列communicationHint)
static <T extends Number> CollectiveGather <T>
createスコープスコープ、オペランド<T>入力、Long groupSize、Long groupKey、Long instanceKey、 Shape shape、 Options ...オプション)
新しいCollectiveGather操作をラップするクラスを作成するファクトリメソッド。
出力<T>
データ()
static CollectiveGather.Options
timeoutSeconds (フロートtimeoutSeconds)

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

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

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

public static CollectiveGather.Options communicationHint (文字列communicationHint)

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

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

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

public Output <T> data ()

public static CollectiveGather.Options timeoutSeconds (Float timeoutSeconds)