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