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.
Métodos Públicos
| Saída <T> | asOutput () Retorna o identificador simbólico de um tensor. |
| estático <T estende número> 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
Métodos Públicos
Saída pública <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 , 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