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> | |
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. |
---|