Aprenda o que há de mais recente em aprendizado de máquina, IA generativa e muito mais no WiML Symposium 2023 Registre-se

Esta página foi traduzida pela API Cloud Translation.

XlaSpmdFullToShardShape

XlaSpmdFullToShardShape classe final pública

Um op usado pelo particionador XLA SPMD para mudar de particionamento automático para

particionamento manual. Ele anota a entrada (forma completa, para ser particionado automaticamente) com o mesmo fragmento usado pelo particionamento manual e gera um tensor em forma de fragmento para ser consumido por operações particionadas manualmente posteriores. Se a forma não puder ser particionada de maneira uniforme, a região de preenchimento será mascarada com 0s.

Constantes

Fragmento OP_NAME O nome desta operação, conforme conhecido pelo motor principal TensorFlow

Métodos Públicos

Output <T>	asOutput () Retorna o identificador simbólico do tensor.
estática <T estende TType > XlaSpmdFullToShardShape <T>	criar ( Scope escopo, Operando <T> de entrada, String manualSharding) Método de fábrica para criar uma classe envolvendo uma nova operação XlaSpmdFullToShardShape.
Output <T>	saída ()

Métodos herdados

De classe org.tensorflow.op.RawOp

final booleano	é igual a (obj Object)
final int	hashCode ()
Operação	op () Retornar esta unidade de computação como uma única `Operation` .
corda final	toString ()

Da classe java.lang.Object

boleano	equals (arg0 Object)
aula final <?>	getClass ()
int	hashCode ()
vazio final	notificar ()
vazio final	notifyAll ()
Fragmento	toString ()
vazio final	wait (longa arg0, int arg1)
vazio final	wait (arg0 de comprimento)
vazio final	wait ()

A partir da interface org.tensorflow.op.Op

abstrato ExecutionEnvironment	env () Retorne o ambiente de execução em que este op foi criado.
abstrato Operação	op () Retornar esta unidade de computação como uma única `Operation` .

A partir da interface org.tensorflow.Operand

abstrato Output <T>	asOutput () Retorna o identificador simbólico do tensor.
T abstrato	asTensor () Retorna o tensor neste operando.
abstrato Forma	forma () Retorna o (possivelmente parcialmente conhecida) forma do tensor referido pela `Output` deste operando.
Classe abstrata <T>	digitar () Retorna o tipo de tensor deste operando

A partir da interface org.tensorflow.ndarray.Shaped

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

Constantes

nome_op final String public static

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

Valor constante: "XlaSpmdFullToShardShape"

Métodos Públicos

pública Output <T> asOutput ()

Retorna o identificador simbólico do 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 XlaSpmdFullToShardShape <T> create ( Scope escopo, Operando <T> de entrada, String manualSharding)

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

Parâmetros

alcance	escopo atual

Devoluções

uma nova instância de XlaSpmdFullToShardShape

XlaSpmdFullToShardShape

Constantes

Métodos Públicos

Métodos herdados

Constantes

nome_op final String public static

Métodos Públicos

pública Output <T> asOutput ()

public static XlaSpmdFullToShardShape <T> create ( Scope escopo, Operando <T> de entrada, String manualSharding)

Parâmetros

Devoluções

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