NcclAllReduce

public final class NcclAllReduce

Produz um tensor contendo a redução em todos os tensores de entrada.

Produz um tensor contendo a redução em todos os tensores de entrada passados ​​para ops dentro do mesmo `shared_name.

O gráfico deve ser construído de forma que se uma operação for executada com o valor shared_name `c`, então as ops` num_devices` serão executadas com o valor shared_name `c`. Caso contrário, 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 realizar. num_devices: o número de dispositivos que participam desta redução. shared_name: Identificador que é compartilhado entre ops da mesma redução.

Métodos Públicos

Output <T>
asOutput ()
Retorna o identificador simbólico de um tensor.
estática <T estende Number> NcclAllReduce <T>
criar ( Scope escopo, Operando <T> entrada, redução String, numDevices longas, String sharedName)
Método de fábrica para criar uma classe que envolve uma nova operação NcclAllReduce.
Output <T>
dados ()

Métodos herdados

Métodos Públicos

pública Output <T> asOutput ()

Retorna o identificador simbólico de um tensor.

As entradas para as 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 ( Scope escopo, Operando <T> entrada, redução String, numDevices longas, 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

pública Output <T> de dados ()