CollectivePermute

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

複製されたTPUインスタンス間でテンソルを並べ替える操作。

各インスタンスは独自の入力を提供します。

たとえば、 `[A、B、C、D]`の4つのTPUインスタンスがあるとします。 source_target_pairs = `[[0,1]、[1,2]、[2,3]、[3,0]]`を渡すと、出力 `[D、A、B、C]`が取得されます。

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリックハンドルを返します。
static <T> CollectivePermute <T>
createスコープスコープ、オペランド<T>入力、オペランド<整数> sourceTargetPairs)
新しいCollectivePermute操作をラップするクラスを作成するファクトリメソッド。
出力<T>
出力()
並べ替えられた入力。

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

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

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

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

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

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

public Output <T> output ()

並べ替えられた入力。