Gera um tensor contendo a redução em todos os tensores de entrada.
Gera um tensor contendo a redução em todos os tensores de entrada passados para operações dentro do mesmo `shared_name.
O gráfico deve ser construído de modo que se uma operação for executada com o valor shared_name `c`, então as operações `num_devices` serão executadas com o valor shared_name `c`. Se isso não for feito, a execução do gráfico não será concluída.
input: a entrada para os dados de redução: o valor da redução em todos os dispositivos `num_devices`. redução: a operação de redução a ser executada. num_devices: O número de dispositivos que participam desta redução. shared_name: Identificador que compartilhou entre as operações da mesma redução.
Métodos públicos
Saída <T> | comoSaída () Retorna o identificador simbólico de um tensor. |
estático <T estende Número> NcclAllReduce <T> | |
Saída <T> | dados () |
Métodos Herdados
Métodos públicos
Public Output <T> asOutput ()
Retorna o identificador simbólico de um tensor.
As entradas para operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.
public static NcclAllReduce <T> create ( Escopo do escopo, Operando <T> entrada, Redução de string, Long numDevices, String sharedName)
Método de fábrica para criar uma classe que envolve uma nova operação NcclAllReduce.
Parâmetros
alcance | escopo atual |
---|
Devoluções
- uma nova instância de NcclAllReduce