publiczna klasa końcowa EnqueueTPUEmbeddingIntegerBatch
Operacja kolejkująca listę wejściowych tensorów wsadowych do TPUEmbedding.
Klasy zagnieżdżone
| klasa | KolejkujTPUEmbeddingIntegerBatch.Options | Opcjonalne atrybuty dla EnqueueTPUEmbeddingIntegerBatch | |
Stałe
| Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
| statyczny EnqueueTPUEmbeddingIntegerBatch | |
| statyczny EnqueueTPUEmbeddingIntegerBatch.Options | urządzenieOrdinal (Długie urządzenieOrdinal) |
Metody dziedziczone
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
public static EnqueueTPUEmbeddingIntegerBatch create ( Zakres zakresu, Iterowalny < Operand < TInt32 >> partia, Operand < TString > modeOverride, Opcje... opcje)
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
public static EnqueueTPUEmbeddingIntegerBatch.Options urządzenieOrdinal (Long DeviceOrdinal)
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ł. |
|---|