NcclReduce

classe final pública NcclReduce

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.

Métodos Públicos

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

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