CollateTPUEmbeddingMemory

kelas akhir publik CollateTPUEmbeddingMemory

Sebuah operasi yang menggabungkan proto konfigurasi memori yang dikodekan string dari semua host.

Metode Publik

Keluaran <String>
sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
CollateTPUEmbeddingMemory statis
buat ( Lingkup lingkup, Iterable< Operand <String>> memoryConfigs)
Metode pabrik untuk membuat kelas yang membungkus operasi CollateTPUEmbeddingMemory baru.
Keluaran <String>

Metode Warisan

Metode Publik

Keluaran publik <String> sebagaiOutput ()

Mengembalikan pegangan simbolik tensor.

Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.

public static CollateTPUEmbeddingMemory buat ( Lingkup lingkup, Iterable< Operand <String>> memoryConfigs)

Metode pabrik untuk membuat kelas yang membungkus operasi CollateTPUEmbeddingMemory baru.

Parameter
cakupan ruang lingkup saat ini
konfigurasi memori Proto konfigurasi memori berkode string yang berisi metadata tentang alokasi memori yang dicadangkan untuk TPUEmbedding di semua host.
Kembali
  • contoh baru dari CollateTPUEmbeddingMemory

Keluaran publik <String> mergeMemoryConfig ()