CollectiveGatherV2

publiczna klasa końcowa CollectiveGatherV2

Wzajemnie gromadzi 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 CollectiveGatherV2 Opcjonalne atrybuty dla CollectiveGatherV2

Metody publiczne

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

public static CollectiveGatherV2 <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ę CollectiveGatherV2.

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

publiczne dane wyjściowe <T> ()

public static CollectiveGatherV2.Options isStateless (boolean isStateless)

publiczny statyczny CollectiveGatherV2.Options limit czasu w sekundach (przekroczenie limitu czasu w sekundach)