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 |
Metody publiczne
statyczny EnqueueTPUEmbeddingIntegerBatch | |
statyczny EnqueueTPUEmbeddingIntegerBatch.Options | urządzenieOrdinal (Długie urządzenieOrdinal) |
Metody dziedziczone
Metody publiczne
public static EnqueueTPUEmbeddingIntegerBatch create ( Zakres zakresu, Iterowalny < Operand <Integer>> partia, Operand <String> 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ł. |
---|