publiczna klasa końcowa TPUPartitionedOutput
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 | TPUPartitionedOutput.Opcje | Opcjonalne atrybuty dla TPUPartitionedOutput | |
Metody publiczne
| statyczne <T> TPUPartitionedOutput <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 TPUPartitionedOutput.Opcje | partycjaDim (Długa partycjaDim) |
Metody dziedziczone
Metody publiczne
public static TPUPartitionedOutput <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ę TPUPartitionedOutput.
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 TPUPartitionedOutput
publiczna Lista< Wyjście <T>> wyjście ()
Lista podzielonych danych wejściowych, które muszą mieć ten sam kształt.
publiczny statyczny TPUPartitionedOutput.Options PartitionDim (Długie partycjiDim)
Parametry
| partycjaWym | Liczba całkowita opisuje, który wymiar jest podzielony. |
|---|