Operacja, która demultipleksuje tensor, który ma zostać podzielony przez XLA na listę partycjonowanych
wyjścia poza obliczeniami XLA. Obsługuje sharding ND.
Metody publiczne
statyczny <T> TPUPartitionedOutputV2 <T> | |
Iterator< Operand <T>> | iterator () |
Lista< Wyjście <T>> | wyjście () Lista podzielonych wyników, które mają ten sam kształt. |
Metody dziedziczone
Metody publiczne
public static TPUPartitionedOutputV2 <T> utwórz (zakres zakresu , wejścia argumentu <T>, Long numSplits, List<Long> partycjiDims)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TPUPartitionedOutputV2.
Parametry
zakres | aktualny zakres |
---|---|
wejścia | Tensor reprezentujący pełny kształt podzielonych tensorów. |
partycjaDims | Lista liczb całkowitych opisujących sposób podziału każdego wymiaru. Pustka wskazuje, że dane wejściowe są replikowane. |
Zwroty
- nowa instancja TPUPartitionedOutputV2