XlaSpmdFullToShardShape

classe final pública XlaSpmdFullToShardShape

Uma operação usada pelo particionador XLA SPMD para mudar do particionamento automático para

particionamento manual. Ele anota a entrada (formato completo, para ser particionado automaticamente) com o mesmo sharding usado pelo particionamento manual e gera um tensor em forma de shard para ser consumido por operações particionadas manualmente posteriormente. Se a forma não for particionável uniformemente, a região de preenchimento será mascarada com 0s.

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 TType > XlaSpmdFullToShardShape <T>	create (escopo do escopo , entrada do operando <T>, string manualSharding) Método de fábrica para criar uma classe que agrupa uma nova operação XlaSpmdFullToShardShape.
Saída <T>	saída ()

Métodos herdados

Da classe org.tensorflow.op.RawOp

booleano final	é igual (objeto obj)
int final	código hash ()
Operação	op () Retorne esta unidade de cálculo como uma única Operation .
sequência final	para sequenciar ()

Da classe java.lang.Object

boleano	é igual (objeto arg0)
aula final<?>	getClass ()
interno	código hash ()
vazio final	notificar ()
vazio final	notificar todos ()
Corda	para sequenciar ()
vazio final	espere (long arg0, int arg1)
vazio final	espere (arg0 longo)
vazio final	espere ()

Da interface org.tensorflow.op.Op

ambiente de execução abstrato	env () Retorne o ambiente de execução em que esta operação foi criada.
operação abstrata	op () Retorne esta unidade de cálculo como uma única Operation .

Da interface org.tensorflow.Operand

saída abstrata <T>	asOutput () Retorna o identificador simbólico do tensor.
abstrato T	comoTensor () Retorna o tensor neste operando.
forma abstrata	forma () Retorna a forma (possivelmente parcialmente conhecida) do tensor referido pela Output deste operando.
Classe abstrata<T>	tipo () Retorna o tipo de tensor deste operando

Da interface org.tensorflow.ndarray.Shaped

abstrato int	classificação ()
forma abstrata	forma ()
abstrato longo	tamanho () Calcula e retorna o tamanho total deste contêiner, em número de valores.

Constantes

String final estática pública OP_NAME

O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow

Valor Constante: "XlaSpmdFullToShardShape"

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 XlaSpmdFullToShardShape <T> create (escopo do escopo , entrada Operand <T>, String manualSharding)

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

Parâmetros

escopo	escopo atual

Devoluções

• uma nova instância de XlaSpmdFullToShardShape

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