CollectiveReduceV2

パブリック最終クラスCollectiveReduceV2

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

「is_stateless」は、各操作が他の集合的な操作への制御依存関係を必要としないことを意味します。この場合、実行時に一意のキー (例: 「instance_key」) を使用して集合グループを区別する必要があります。

ネストされたクラス

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

パブリックメソッド

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

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

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

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

public static CollectiveReduceV2.Options communicationHint (String communicationHint)

public static CollectiveReduceV2 <T> create ( Scopeスコープ、オペランド<T> input、オペランド<Integer> groupSize、オペランド<Integer> groupKey、オペランド<Integer> インスタンスキー、Iterable< Operand <?>> orderToken、String mergeOp、String FinalOp、オプション...オプション)

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

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

public出力<T>データ()

public static CollectiveReduceV2.Options isStateless (Boolean isStateless)

public static CollectiveReduceV2.Options maxSubdivsPerDevice (Long maxSubdivsPerDevice)

public static CollectiveReduceV2.Options timeoutSeconds (Float timeoutSeconds)