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

Metody publiczne

statyczny EnqueueTPUEmbeddingIntegerBatch	utwórz (zakres zakresu , Iterable< Operand <Integer>> partia, Operand <String> 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

Z klasy org.tensorflow.op.PrimitiveOp

końcowa wartość logiczna	równa się (obiekt obiektu)
końcowy wew	hashCode ()
Operacja	op () Zwraca podstawową Operation
ostatni ciąg	doString ()

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ć ()

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