パブリック最終クラスCollectivePermute
レプリケートされた TPU インスタンス全体でテンソルを並べ替える Op。
各インスタンスは独自の入力を提供します。
たとえば、「[A, B, C, D]」という 4 つの TPU インスタンスがあるとします。 source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` を渡すと、出力 `[D, A, B, C]` が取得されます。
パブリックメソッド
出力<T> | asOutput () テンソルのシンボリック ハンドルを返します。 |
静的 <T> CollectivePermute <T> | |
出力<T> | 出力() 並べ替えられた入力。 |
継承されたメソッド
パブリックメソッド
public Output <T> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static CollectivePermute <T> create (スコープscope、オペランド<T>入力、オペランド<Integer>sourceTargetPairs)
新しい CollectivePermute オペレーションをラップするクラスを作成するためのファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
入力 | 並べ替えられるローカル入力。現在、float と bfloat16 のみをサポートしています。 |
ソースターゲットペア | 形状 [num_pairs, 2] のテンソル。 |
戻り値
- CollectivePermute の新しいインスタンス