CollectiveReduceV2

publiczna klasa końcowa CollectiveReduceV2

Wzajemnie redukuje wiele tensorów tego samego typu i kształtu.

„is_stateless” oznacza, że ​​każda operacja nie potrzebuje zależności kontrolnych od innych operacji zbiorowych. W takim przypadku do rozróżnienia grup zbiorczych należy użyć kluczy, które są unikalne w czasie wykonywania (np. `klucz_instancji`).

Klasy zagnieżdżone

klasa Opcje zbiorczeReduceV2 Opcjonalne atrybuty dla CollectiveReduceV2

Metody publiczne

Wyjście <T>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczne opcje CollectiveReduceV2
communicationHint (ciąg znaków communicationHint)
statyczny <T numer rozszerzenia> CollectiveReduceV2 <T>
utwórz ( Zakres zakresu, Operand <T> input, Operand <Integer> groupSize, Operand <Integer> groupKey, Operand <Integer> instanKey, Iterable< Operand <?>> OrderingToken, String mergeOp, String finalOp, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację CollectiveReduceV2.
Wyjście <T>
dane ()
statyczne opcje CollectiveReduceV2
isStateless (boolean isStateless)
statyczne opcje CollectiveReduceV2
maxSubdivsPerDevice (Długie maxSubdivsPerDevice)
statyczne opcje CollectiveReduceV2
timeoutSeconds (limit czasu pływania w sekundach)

Metody dziedziczone

Metody publiczne

publiczne wyjście <T> asOutput ()

Zwraca symboliczny uchwyt tensora.

Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.

public static CollectiveReduceV2.Options communicationHint (String communicationHint)

public static CollectiveReduceV2 <T> create ( Zakres zakresu , Operand <T> wejście, Operand <Integer> groupSize, Operand <Integer> groupKey, Operand <Integer> instanKey, Iterable< Operand <?>> OrderingToken, String mergeOp, String finalOp, Opcje... opcje)

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację CollectiveReduceV2.

Parametry
zakres aktualny zakres
opcje przenosi opcjonalne wartości atrybutów
Zwroty
  • nowa instancja CollectiveReduceV2

publiczne dane wyjściowe <T> ()

public static CollectiveReduceV2.Options isStateless (boolean isStateless)

public static CollectiveReduceV2.Options maxSubdivsPerDevice (Długie maxSubdivsPerDevice)

public static CollectiveReduceV2.Options timeoutSeconds (Float timeoutSeconds)