CollateTPUEmbeddingMemory

publiczna klasa końcowa CollateTPUEmbeddingMemory

Operacja, która łączy protosy konfiguracji pamięci zakodowanej w postaci ciągów znaków ze wszystkich hostów.

Metody publiczne

Wyjście <Ciąg>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczny SortujTPUEmbeddingMemory
utwórz (zakres zakresu , Iterable< Operand <String>> memoryConfigs)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację CollateTPUEmbeddingMemory.
Wyjście <Ciąg>

Metody dziedziczone

Metody publiczne

publiczne wyjście <String> asOutput ()

Zwraca symboliczny uchwyt tensora.

Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.

public static CollateTPUEmbeddingMemory create ( Zakres zakresu, Iterowalny < Operand <String>> memoryConfigs)

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację CollateTPUEmbeddingMemory.

Parametry
zakres aktualny zakres
Konfiguracje pamięci Protokoły konfiguracji pamięci zakodowane w postaci ciągów znaków zawierające metadane dotyczące alokacji pamięci zarezerwowanych dla osadzania TPU na wszystkich hostach.
Zwroty
  • nowa instancja CollateTPUEmbeddingMemory

publiczne wyjście <String> mergedMemoryConfig ()