CollectivePermute

パブリック最終クラス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>
create (スコープscope、オペランド<T>入力、オペランド<Integer>sourceTargetPairs)
新しい CollectivePermute オペレーションをラップするクラスを作成するためのファクトリ メソッド。
出力<T>
出力()
並べ替えられた入力。

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

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

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

public static CollectivePermute <T> create (スコープscope、オペランド<T>入力、オペランド<Integer>sourceTargetPairs)

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

パラメーター
範囲現在のスコープ
入力並べ替えられるローカル入力。現在、float と bfloat16 のみをサポートしています。
ソースターゲットペア形状 [num_pairs, 2] のテンソル。
戻り値
  • CollectivePermute の新しいインスタンス

public出力<T>出力()

並べ替えられた入力。