TPUReplicatedInput

classe final pública TPUReplicatedInput

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

Esta operação mantém uma entrada replicada para um subgráfico de computação `tpu.replicate ()`. Cada entrada replicada tem a mesma forma e tipo junto com a saída.

Por exemplo:

%a = "tf.opA"()
 %b = "tf.opB"()
 %replicated_input = "tf.TPUReplicatedInput"(%a, %b)
 %computation = "tf.Computation"(%replicated_input)
 
O cálculo 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>
asOutput ()
Retorna o identificador simbólico de um tensor.
static <T> TPUReplicatedInput <T>
criar ( escopo do escopo, entradas Iteráveis ​​< Operando <T>>, Opções ... opções)
Método de fábrica para criar uma classe que envolve uma nova operação TPUReplicatedInput.
TPUReplicatedInput.Options estáticos
índice ( índice longo)
TPUReplicatedInput.Options estáticos
isMirroredVariable (Boolean isMirroredVariable)
TPUReplicatedInput.Options estáticos
isPacked (Boolean 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 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 TPUReplicatedInput <T> create ( Scope escopo, Iterable < Operando <T >> entradas, Options ... 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

public static TPUReplicatedInput.Options index (Long index)

public static TPUReplicatedInput.Options isMirroredVariable (Boolean isMirroredVariable)

public static TPUReplicatedInput.Options isPacked (Boolean isPacked)

output public Output <T> output ()