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>>	iterator ()
Lista< Wyjście <T>>	wyjście () Lista podzielonych wyników, które mają ten sam kształt.

Metody dziedziczone

Z klasy org.tensorflow.op.PrimitiveOp

końcowa wartość logiczna	równa się (obiekt obiektu)
końcowy wew	hashCode ()
Operacja	op () Zwraca podstawową Operation
ostatni ciąg	doString ()

Z klasy java.lang.Object

wartość logiczna	równa się (Obiekt arg0)
ostatnia klasa<?>	pobierzKlasę ()
wew	hashCode ()
ostateczna pustka	powiadomić ()
ostateczna pustka	powiadom wszystkich ()
Strunowy	doString ()
ostateczna pustka	czekaj (długi arg0, int arg1)
ostateczna pustka	czekaj (długi arg0)
ostateczna pustka	Czekać ()

Z interfejsu java.lang.Iterable

próżnia	forEach (Konsument<? super T> arg0)
abstrakcyjny Iterator< Operand <T>>	iterator ()
Rozdzielacz< Operand <T>>	rozdzielacz ()

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.