PartitionedOutput

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

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.

public static PartitionedOutput.Options partycjaDim (Długa partycjaDim)

Parametry
partycjaWym Liczba całkowita opisuje, który wymiar jest podzielony.