CollectiveAllToAllV2

パブリック最終クラスCollectiveAllToAllV2

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

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

ネストされたクラス

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

パブリックメソッド

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

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

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

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

public static CollectiveAllToAllV2.Options communicationHint (String communicationHint)

public static CollectiveAllToAllV2 <T> create (スコープscope、オペランド<T>入力、オペランド<Integer>groupSize、オペランド<Integer>groupKey、オペランド<Integer>instanceKey、Iterable< Operand <?>>orderingToken、 Options...オプション)

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

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

public出力<T>データ()

public static CollectiveAllToAllV2.Options isStateless (Boolean isStateless)

public static CollectiveAllToAllV2.Options timeoutSeconds (Float timeoutSeconds)