NcclReduce

public final class NcclReduce

Reduz a `entrada` de` num_devices` usando `redução` para um único dispositivo.

Reduz a `entrada` de` num_devices` usando `redução` 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 o próprio op seja atribuído 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 realizar.

Métodos Públicos

Output <T>
asOutput ()
Retorna o identificador simbólico de um tensor.
estática <T estende Number> NcclReduce <T>
criar ( Âmbito âmbito, Iterable < Operando <t >> entrada, redução String)
Método de fábrica para criar uma classe envolvendo uma nova operação NcclReduce.
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 NcclReduce <T> create ( Scope escopo, Iterable < Operando <T >> entrada, redução String)

Método de fábrica para criar uma classe envolvendo uma nova operação NcclReduce.

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

pública Output <T> de dados ()