Reduz `input` de `num_devices` usando `reduction` para um único dispositivo.
Reduz `input` de `num_devices` usando `reduction` para um único dispositivo.
O gráfico deve ser construído de forma que todas as entradas tenham uma atribuição de dispositivo válida e a própria operação seja atribuída a um desses dispositivos.
entrada: A entrada para a redução. data: o valor da redução em todos os dispositivos `num_devices`. redução: a operação de redução a ser executada.
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 > NcclReduce <T> | |
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
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 NcclReduce <T> create (escopo do escopo , entrada Iterable< Operand <T>>, redução de string)
Método de fábrica para criar uma classe que envolve uma nova operação NcclReduce.
Parâmetros
escopo | escopo atual |
---|
Devoluções
- uma nova instância de NcclReduce