TPUPartitionedOutputV2

publiczna klasa końcowa TPUPartitionedOutputV2

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>
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.
Iterator< Operand <T>>
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

publiczny Iterator< Operand <T>> iterator ()

publiczna Lista< Wyjście <T>> wyjście ()

Lista podzielonych wyników, które mają ten sam kształt.