EnqueueTPUEmbeddingIntegerBatch
Zadbaj o dobrą organizację dzięki kolekcji
Zapisuj i kategoryzuj treści zgodnie ze swoimi preferencjami.
Operacja kolejkująca listę wejściowych tensorów wsadowych do TPUEmbedding.
Stałe
Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody dziedziczone
Z klasy java.lang.Object wartość logiczna | równa się (Obiekt arg0) |
ostatnia klasa<?> | pobierzKlasę () |
wew | hashCode () |
ostateczna pustka | powiadomić () |
ostateczna pustka | powiadom wszystkich () |
Strunowy | doString () |
ostateczna pustka | czekaj (długi arg0, int arg1) |
ostateczna pustka | czekaj (długi arg0) |
ostateczna pustka | Czekać () |
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Wartość stała: „EnqueueTPUEmbeddingIntegerBatch”
Metody publiczne
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację EnqueueTPUEmbeddingIntegerBatch.
Parametry
zakres | aktualny zakres |
---|
seria | Lista tensorów 1D, po jednym dla każdej tabeli osadzania, zawierająca indeksy do tabel. |
---|
trybZastąpienie | Ciąg wejściowy, który zastępuje tryb określony w TPUEmbeddingConfiguration. Obsługiwane wartości to {'nieokreślone', 'wnioskowanie', 'trening', 'backward_pass_only'}. W przypadku ustawienia na „unspecified” używany jest tryb ustawiony w TPUEmbeddingConfiguration, w przeciwnym razie używany jest mode_override. |
---|
opcje | przenosi opcjonalne wartości atrybutów |
---|
Zwroty
- nowa instancja EnqueueTPUEmbeddingIntegerBatch
Parametry
urządzeniePorządek | Urządzenie TPU, którego chcesz użyć. Powinna wynosić >= 0 i być mniejsza od liczby rdzeni TPU w zadaniu, na którym umieszczony jest węzeł. |
---|
O ile nie stwierdzono inaczej, treść tej strony jest objęta licencją Creative Commons – uznanie autorstwa 4.0, a fragmenty kodu są dostępne na licencji Apache 2.0. Szczegółowe informacje na ten temat zawierają zasady dotyczące witryny Google Developers. Java jest zastrzeżonym znakiem towarowym firmy Oracle i jej podmiotów stowarzyszonych.
Ostatnia aktualizacja: 2025-07-26 UTC.
[null,null,["Ostatnia aktualizacja: 2025-07-26 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|---------------|-----------------------------------------------------------------------------------------------------------------------|"]]