publiczna klasa końcowa PartitionedOutput
Operacja, która demultipleksuje tensor, który ma zostać podzielony przez XLA na listę partycjonowanych
wyjścia poza obliczeniami XLA.
Klasy zagnieżdżone
klasa | PartitionedOutput.Opcje | Opcjonalne atrybuty dla PartitionedOutput |
Stałe
Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
statyczny <T rozszerza TType > PartitionedOutput <T> | |
Iterator< Operand <T>> | iterator () |
Lista< Wyjście <T>> | wyjście () Lista podzielonych danych wejściowych, które muszą mieć ten sam kształt. |
statyczne PartitionedOutput.Options | partycjaDim (Długa partycjaDim) |
Metody dziedziczone
wartość logiczna | równa się (Obiekt arg0) |
ostatnia klasa<?> | pobierzKlasę () |
wew | hashCode () |
ostateczna pustka | powiadomić () |
ostateczna pustka | powiadom wszystkich () |
Strunowy | doString () |
ostateczna pustka | czekaj (długi arg0, int arg1) |
ostateczna pustka | czekaj (długi arg0) |
ostateczna pustka | Czekać () |
abstrakcyjne środowisko wykonania | środowisko () Zwróć środowisko wykonawcze, w którym utworzono tę operację. |
abstrakcyjna operacja |
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Wartość stała: „TPUPartitionedOutput”
Metody publiczne
public static PartitionedOutput <T> utwórz (zakres zakresu , wejścia argumentu <T>, długie numSplits, opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację PartitionedOutput.
Parametry
zakres | aktualny zakres |
---|---|
wejścia | Tensor reprezentujący pełny kształt podzielonych tensorów. |
opcje | przenosi opcjonalne wartości atrybutów |
Zwroty
- nowa instancja PartitionedOutput
publiczna Lista< Wyjście <T>> wyjście ()
Lista podzielonych danych wejściowych, które muszą mieć ten sam kształt.
public static PartitionedOutput.Options partycjaDim (Długa partycjaDim)
Parametry
partycjaWym | Liczba całkowita opisuje, który wymiar jest podzielony. |
---|