CollectiveReduceScatterV2

publiczna klasa końcowa CollectiveReduceScatterV2

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

„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 zbiorczeReduceScatterV2 Opcjonalne atrybuty dla CollectiveReduceScatterV2

Metody publiczne

Wyjście <T>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczne CollectiveReduceScatterV2.Opcje
communicationHint (ciąg znaków communicationHint)
statyczny <T numer rozszerzenia> CollectiveReduceScatterV2 <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ę CollectiveReduceScatterV2.
Wyjście <T>
dane ()
statyczne CollectiveReduceScatterV2.Opcje
isStateless (boolean isStateless)
statyczne CollectiveReduceScatterV2.Opcje
maxSubdivsPerDevice (Długie maxSubdivsPerDevice)
statyczne CollectiveReduceScatterV2.Opcje
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 CollectiveReduceScatterV2.Options communicationHint (String communicationHint)

public static CollectiveReduceScatterV2 <T> create ( Zakres zasięgu, 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ę CollectiveReduceScatterV2.

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

publiczne dane wyjściowe <T> ()

public static CollectiveReduceScatterV2.Options isStateless (Boolean isStateless)

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

public static CollectiveReduceScatterV2.Options timeoutSeconds (Float timeoutSeconds)