NcclReduce

publiczna klasa końcowa NcclReduce

Zmniejsza „wejście” z „liczby_urządzeń” za pomocą „redukcji” do jednego urządzenia.

Zmniejsza „wejście” z „liczby_urządzeń” za pomocą „redukcji” do jednego urządzenia.

Wykres powinien być tak skonstruowany, aby wszystkie wejścia miały prawidłowe przypisanie urządzenia, a sam proces miał przypisane jedno z tych urządzeń.

wejście: wejście do redukcji. dane: wartość redukcji na wszystkich urządzeniach `num_devices`. redukcja: operacja redukcji do wykonania.

Metody publiczne

Wyjście <T>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczny <T numer rozszerzenia> NcclReduce <T>
utwórz (zakres zakresu , wejście Iterable< Operand <T>>, redukcja ciągu)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację NcclReduce.
Wyjście <T>
dane ()

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 NcclReduce <T> create (zakres zakresu , wejście Iterable< Operand <T>>, redukcja ciągu znaków)

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

Parametry
zakres aktualny zakres
Zwroty
  • nowa instancja NcclReduce

publiczne dane wyjściowe <T> ()