TPUReplicatedInput

classe final pública TPUReplicatedInput

Conecta N entradas a uma computação TPU replicada de N-way.

Esta operação contém uma entrada replicada para um subgrafo de computação `tpu.replicate()`. Cada entrada replicada tem a mesma forma e tipo ao lado da saída.

Por exemplo:

%a = "tf.opA"()
 %b = "tf.opB"()
 %replicated_input = "tf.TPUReplicatedInput"(%a, %b)
 %computation = "tf.Computation"(%replicated_input)
 
A computação acima tem uma entrada replicada de duas réplicas.

Classes aninhadas

aula TPUReplicatedInput.Options Atributos opcionais para TPUReplicatedInput

Métodos públicos

Saída <T>
comoSaída ()
Retorna o identificador simbólico de um tensor.
estático <T> TPUReplicatedInput <T>
create ( Escopo do escopo, entradas Iterable< Operando <T>>, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação TPUReplicatedInput.
static TPUReplicatedInput.Options
índice (índice longo)
static TPUReplicatedInput.Options
isMirroredVariable (Boolean isMirroredVariable)
static TPUReplicatedInput.Options
isPacked (Booleano isPacked)
Saída <T>
saída ()

Métodos Herdados

Métodos públicos

Public Output <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 TPUReplicatedInput <T> create ( Escopo do escopo, Iterable< Operando <T>> entradas, Opções... opções)

Método de fábrica para criar uma classe que envolve uma nova operação TPUReplicatedInput.

Parâmetros
alcance escopo atual
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de TPUReplicatedInput

índice público estático TPUReplicatedInput.Options ( índice longo)

public static TPUReplicatedInput.Options isMirroredVariable (Boolean isMirroredVariable)

public static TPUReplicatedInput.Options isPacked (Boolean isPacked)

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