TPUPartitionedInputV2

publiczna klasa końcowa TPUPartitionedInputV2

Operacja grupująca listę podzielonych danych wejściowych. Obsługuje sharding ND.

Klasy zagnieżdżone

klasa Opcje TPUPartitionedInputV2 Opcjonalne atrybuty dla TPUPartitionedInputV2

Metody publiczne

Wyjście <T>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczny <T> TPUPartitionedInputV2 <T>
utwórz ( Zakres zakresu, wejścia Iterable< Operand <T>>, List<Long> partycjeDims, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TPUPartitionedInputV2.
statyczne TPUPartitionedInputV2.Opcje
isPacked (boolean isPacked)
Wyjście <T>
wyjście ()
Uchwyt reprezentujący pełny kształt podzielonych tensorów.

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 TPUPartitionedInputV2 <T> utwórz (zakres zakresu , wejścia Iterable< Operand <T>>, List<Long> partycjeDims, Opcje... opcje)

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TPUPartitionedInputV2.

Parametry
zakres aktualny zakres
wejścia Lista podzielonych danych wejściowych, które muszą mieć ten sam kształt.
partycjaDims Lista liczb całkowitych opisujących sposób podziału każdego wymiaru. Pustka wskazuje, że dane wejściowe są replikowane.
opcje przenosi opcjonalne wartości atrybutów
Zwroty
  • nowa instancja TPUPartitionedInputV2

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

Parametry
Jest zapakowane Wskazuje, czy dane wejściowe są zasobem spakowanym.

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

Uchwyt reprezentujący pełny kształt podzielonych tensorów.