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.
Constantes
Fragmento | OP_NAME | O nome desta operação, conforme conhecido pelo motor principal TensorFlow |
Métodos Públicos
Output <T> | asOutput () Retorna o identificador simbólico do tensor. |
estática <T estende TNumber > NcclAllReduce <T> | |
Output <T> | dados () |
Métodos herdados
Constantes
nome_op final String public static
O nome desta operação, conforme conhecido pelo motor principal TensorFlow
Métodos Públicos
pública Output <T> asOutput ()
Retorna o identificador simbólico do 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