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ł. |
|---|