NcclAllReduce

classe final pública 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 operações dentro do mesmo `shared_name.

O gráfico deve ser construído de forma que se uma operação for executada com valor de nome_compartilhado `c`, então as operações `num_devices` serão executadas com valor de nome_compartilhado `c`. Não fazer isso fará com que a execução do gráfico não seja concluída.

entrada: 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 participantes desta redução. shared_name: Identificador compartilhado entre operações da mesma redução.

Constantes

Corda OP_NAME O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow

Métodos Públicos

Saída <T>
asOutput ()
Retorna o identificador simbólico do tensor.
estático <T estende TNumber > NcclAllReduce <T>
create (escopo do escopo , entrada do operando <T>, redução de string, numDevices longos, string sharedName)
Método de fábrica para criar uma classe que envolve uma nova operação NcclAllReduce.
Saída <T>
dados ()

Métodos herdados

Constantes

String final estática pública OP_NAME

O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow

Valor constante: "NcclAllReduce"

Métodos Públicos

Saída pública <T> asOutput ()

Retorna o identificador simbólico do 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 , entrada do operando <T>, redução de string, numDevices longos, string sharedName)

Método de fábrica para criar uma classe que envolve uma nova operação NcclAllReduce.

Parâmetros
escopo escopo atual
Devoluções
  • uma nova instância de NcclAllReduce

dados de saída pública <T> ()