EnqueueTPUEmbeddingIntegerBatch

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
utwórz ( zakres zakresu, Iterable< Operand < TInt32 >> partia, Operand < TString > modeOverride, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację 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ł.