TPUReplicatedInput

publiczna klasa końcowa TPUReplicatedInput

Łą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 obliczenie obejmuje zreplikowane dane wejściowe w postaci dwóch replik.

Klasy zagnieżdżone

klasa Opcje replikowanego wejścia TPU Opcjonalne atrybuty dla TPUReplicatedInput

Metody publiczne

Wyjście <T>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczne <T> TPUReplikowane wejście <T>
utwórz (zakres zakresu , wejścia Iterable< Operand <T>>, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TPUReplicatedInput.
statyczne TPUReplicatedInput.Options
indeks (długi indeks)
statyczne TPUReplicatedInput.Options
isMirroredVariable (Boolean isMirroredVariable)
statyczne TPUReplicatedInput.Options
isPacked (boolean isPacked)
Wyjście <T>

Metody dziedziczone

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
Zwroty
  • nowa instancja TPUReplicatedInput

publiczny statyczny indeks TPUReplicatedInput.Options (indeks długi)

public static TPUReplicatedInput.Options isMirroredVariable (Boolean isMirroredVariable)

publiczny statyczny TPUReplicatedInput.Options jest spakowany (boolean jest spakowany)

publiczne wyjście <T> wyjście ()