publiczna klasa końcowa TPUReplicatedOutput
Łączy N wyjść z N-kierunkowo replikowanych obliczeń TPU.
Ta operacja przechowuje zreplikowany wynik z podgrafu obliczeniowego `tpu.replicate()`. Każdy zreplikowany wynik ma ten sam kształt i typ wraz z wejściem.
Na przykład:
%computation = "tf.Computation"()
%replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)
Powyższe obliczenie daje zreplikowany wynik w postaci dwóch replik. Metody publiczne
statyczne <T> TPUReplikowane wyjście <T> | |
Iterator< Operand <T>> | iterator () |
Lista< Wyjście <T>> | wyjścia () |
Metody dziedziczone
Metody publiczne
public static TPUReplicatedOutput <T> utwórz (zakres zakresu , dane wejściowe argumentu <T>, Long numReplicas)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TPUReplicatedOutput.
Parametry
zakres | aktualny zakres |
---|
Zwroty
- nowa instancja TPUReplicatedOutput