TPUPartitionedOutput

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

publiczny Iterator< Operand <T>> iterator ()

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.