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