Łączy N wejść z N-kierunkowo replikowanymi obliczeniami TPU.
Ta operacja przechowuje zreplikowane dane wejściowe w podgrafie obliczeniowym `tpu.replicate()`. Każde zreplikowane dane wejściowe mają ten sam kształt i typ wraz z danymi wyjściowymi.
Na przykład:
%a = "tf.opA"()
%b = "tf.opB"()
%replicated_input = "tf.TPUReplicatedInput"(%a, %b)
%computation = "tf.Computation"(%replicated_input)
Powyższe obliczenia obejmują zreplikowane dane wejściowe w postaci dwóch replik. Klasy zagnieżdżone
klasa | Opcje replikowanego wejścia TPU | Opcjonalne atrybuty dla TPUReplicatedInput |
Stałe
Smyczkowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
Wyjście <T> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
statyczne <T rozszerza TType > TPUReplicatedInput <T> | |
statyczne TPUReplicatedInput.Options | indeks (długi indeks) |
statyczne TPUReplicatedInput.Options | isMirroredVariable (Boolean isMirroredVariable) |
statyczne TPUReplicatedInput.Options | isPacked (boolean isPacked) |
Wyjście <T> | wyjście () |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Metody publiczne
publiczne wyjście <T> asOutput ()
Zwraca symboliczny uchwyt tensora.
Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.
public static TPUReplicatedInput <T> create (zakres zakresu , wejścia Iterable< Operand <T>>, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TPUReplicatedInput.
Parametry
zakres | aktualny zakres |
---|---|
opcje | przenosi opcjonalne wartości atrybutów |
Powroty
- nowa instancja TPUReplicatedInput