CollectiveAllToAllV2

publiczna klasa końcowa CollectiveAllToAllV2

Wzajemnie wymienia 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 zbiorczeAllToAllV2 Opcjonalne atrybuty dla CollectiveAllToAllV2

Metody publiczne

Wyjście <T>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczne CollectiveAllToAllV2.Options
communicationHint (ciąg znaków communicationHint)
statyczny <T numer rozszerzenia> CollectiveAllToAllV2 <T>
utwórz (Zakres zakresu , Operand <T> dane wejściowe, Operand <Integer> groupSize, Operand <Integer> groupKey, Operand <Integer> instanKey, Iterable< Operand <?>> OrderingToken, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację CollectiveAllToAllV2.
Wyjście <T>
dane ()
statyczne CollectiveAllToAllV2.Options
isStateless (boolean isStateless)
statyczne CollectiveAllToAllV2.Options
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 CollectiveAllToAllV2.Options communicationHint (String communicationHint)

public static CollectiveAllToAllV2 <T> create (Zakres zakresu , Operand <T> dane wejściowe, Operand <Integer> groupSize, Operand <Integer> groupKey, Operand <Integer> instanKey, Iterable< Operand <?>> OrderingToken, Opcje... opcje)

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

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

publiczne dane wyjściowe <T> ()

public static CollectiveAllToAllV2.Options isStateless (boolean isStateless)

public static CollectiveAllToAllV2.Options timeoutSeconds (Float timeoutSeconds)