パブリック最終クラスTPUReplicatedOutput
N ウェイ複製された TPU 計算からの N 出力を接続します。
この操作は、「tpu.replicate()」計算サブグラフからの複製された出力を保持します。複製された各出力は、入力と同じ形状とタイプを持ちます。
例えば:
%computation = "tf.Computation"()
%replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)
上記の計算では、2 つのレプリカの複製出力が得られます。 パブリックメソッド
静的 <T> TPUReplicatedOutput <T> | |
イテレータ<オペランド<T>> | イテレータ() |
リスト<出力<T>> | 出力() |
継承されたメソッド
パブリックメソッド
public static TPUReplicatedOutput <T> create (スコープscope、オペランド<T>入力、Long numReplicas)
新しい TPUReplicatedOutput オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメータ
範囲 | 現在のスコープ |
---|
返品
- TPUReplicatedOutput の新しいインスタンス