Aprenda o que há de mais recente em aprendizado de máquina, IA generativa e muito mais no WiML Symposium 2023
Registre-se
EnqueueTPUEmbeddingIntegerBatch
Mantenha tudo organizado com as coleções
Salve e categorize o conteúdo com base nas suas preferências.
Um op que enfileira uma lista de tensores de lote de entrada para TPUEmbedding.
Constantes
| Fragmento | OP_NAME | O nome desta operação, conforme conhecido pelo motor principal TensorFlow |
Métodos herdados
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 () |
Constantes
nome_op final String public static
O nome desta operação, conforme conhecido pelo motor principal TensorFlow
Valor constante: "EnqueueTPUEmbeddingIntegerBatch"
Métodos Públicos
Método de fábrica para criar uma classe que envolve uma nova operação EnqueueTPUEmbeddingIntegerBatch.
Parâmetros
| alcance | escopo atual |
|---|
| lote | Uma lista de tensores 1D, um para cada tabela de incorporação, contendo os índices nas tabelas. |
|---|
| modeOverride | Uma entrada de string que substitui o modo especificado na TPUEmbeddingConfiguration. Os valores suportados são {'não especificado', 'inferência', 'treinamento', 'backward_pass_only'}. Quando definido como 'não especificado', o modo definido em TPUEmbeddingConfiguration é usado, caso contrário, mode_override é usado. |
|---|
| opções | carrega valores de atributos opcionais |
|---|
Devoluções
- uma nova instância de EnqueueTPUEmbeddingIntegerBatch
Parâmetros
| deviceOrdinal | O dispositivo TPU a ser usado. Deve ser> = 0 e menor que o número de núcleos de TPU na tarefa em que o nó está colocado. |
|---|
Exceto em caso de indicação contrária, o conteúdo desta página é licenciado de acordo com a Licença de atribuição 4.0 do Creative Commons, e as amostras de código são licenciadas de acordo com a Licença Apache 2.0. Para mais detalhes, consulte as políticas do site do Google Developers. Java é uma marca registrada da Oracle e/ou afiliadas.
Última atualização 2021-11-29 UTC.
[null,null,["Última atualização 2021-11-29 UTC."],[],[],null,["# EnqueueTPUEmbeddingIntegerBatch\n\npublic final class **EnqueueTPUEmbeddingIntegerBatch** \nAn op that enqueues a list of input batch tensors to TPUEmbedding. \n\n### Nested Classes\n\n|-------|---|---|-------------------------------------------------------------------------------------------------------------------------------------|\n| class | [EnqueueTPUEmbeddingIntegerBatch.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.Options) || Optional attributes for [EnqueueTPUEmbeddingIntegerBatch](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch) |\n\n### Constants\n\n|--------|---------------------------------------------------------------------------------------------|---------------------------------------------------------|\n| String | [OP_NAME](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch#OP_NAME) | The name of this op, as known by TensorFlow core engine |\n\n### Public Methods\n\n|------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| static [EnqueueTPUEmbeddingIntegerBatch](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch) | [create](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch#create(org.tensorflow.op.Scope, java.lang.Iterable\u003corg.tensorflow.Operand\u003corg.tensorflow.types.TInt32\u003e\u003e, org.tensorflow.Operand\u003corg.tensorflow.types.TString\u003e, org.tensorflow.op.tpu.EnqueueTPUEmbeddingIntegerBatch.Options...))([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TInt32](/jvm/api_docs/java/org/tensorflow/types/TInt32)\\\u003e\\\u003e batch, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TString](/jvm/api_docs/java/org/tensorflow/types/TString)\\\u003e modeOverride, [Options...](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.Options) options) Factory method to create a class wrapping a new EnqueueTPUEmbeddingIntegerBatch operation. |\n| static [EnqueueTPUEmbeddingIntegerBatch.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.Options) | [deviceOrdinal](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch#deviceOrdinal(java.lang.Long))(Long deviceOrdinal) |\n\n### Inherited Methods\n\nFrom class [org.tensorflow.op.RawOp](/jvm/api_docs/java/org/tensorflow/op/RawOp) \n\n|----------------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| final boolean | [equals](/jvm/api_docs/java/org/tensorflow/op/RawOp#equals(java.lang.Object))(Object obj) |\n| final int | [hashCode](/jvm/api_docs/java/org/tensorflow/op/RawOp#hashCode())() |\n| [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/RawOp#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n| final String | [toString](/jvm/api_docs/java/org/tensorflow/op/RawOp#toString())() |\n\nFrom class java.lang.Object \n\n|------------------|---------------------------|\n| boolean | equals(Object arg0) |\n| final Class\\\u003c?\\\u003e | getClass() |\n| int | hashCode() |\n| final void | notify() |\n| final void | notifyAll() |\n| String | toString() |\n| final void | wait(long arg0, int arg1) |\n| final void | wait(long arg0) |\n| final void | wait() |\n\nFrom interface [org.tensorflow.op.Op](/jvm/api_docs/java/org/tensorflow/op/Op) \n\n|-----------------------------------------------------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| abstract [ExecutionEnvironment](/jvm/api_docs/java/org/tensorflow/ExecutionEnvironment) | [env](/jvm/api_docs/java/org/tensorflow/op/Op#env())() Return the execution environment this op was created in. |\n| abstract [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/Op#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n\nConstants\n---------\n\n#### public static final String\n**OP_NAME**\n\nThe name of this op, as known by TensorFlow core engine \nConstant Value: \"EnqueueTPUEmbeddingIntegerBatch\"\n\nPublic Methods\n--------------\n\n#### public static [EnqueueTPUEmbeddingIntegerBatch](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch)\n**create**\n([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TInt32](/jvm/api_docs/java/org/tensorflow/types/TInt32)\\\u003e\\\u003e batch, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TString](/jvm/api_docs/java/org/tensorflow/types/TString)\\\u003e modeOverride, [Options...](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.Options) options)\n\nFactory method to create a class wrapping a new EnqueueTPUEmbeddingIntegerBatch operation. \n\n##### Parameters\n\n| scope | current scope |\n| batch | A list of 1D tensors, one for each embedding table, containing the indices into the tables. |\n| modeOverride | A string input that overrides the mode specified in the TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference', 'training', 'backward_pass_only'}. When set to 'unspecified', the mode set in TPUEmbeddingConfiguration is used, otherwise mode_override is used. |\n| options | carries optional attributes values |\n|--------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n\n##### Returns\n\n- a new instance of EnqueueTPUEmbeddingIntegerBatch \n\n#### public static [EnqueueTPUEmbeddingIntegerBatch.Options](/jvm/api_docs/java/org/tensorflow/op/tpu/EnqueueTPUEmbeddingIntegerBatch.Options)\n**deviceOrdinal**\n(Long deviceOrdinal)\n\n\u003cbr /\u003e\n\n##### Parameters\n\n| deviceOrdinal | The TPU device to use. Should be \\\u003e= 0 and less than the number of TPU cores in the task on which the node is placed. |\n|---------------|-----------------------------------------------------------------------------------------------------------------------|"]]
