TPUOrdinalSelector

TPUOrdinalSelector kelas akhir publik

Operasi pemilih inti TPU.

Operasi ini menghasilkan satu set inti TPU (untuk pemanasan) atau satu inti TPU (untuk inferensi reguler) untuk menjalankan program TPU. Outputnya dikonsumsi oleh TPUPartitionedCall.

Metode Publik

Keluaran <Bilangan Bulat>
sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
TPUOrdinalSelector statis
membuat ( Lingkup ruang lingkup)
Metode pabrik untuk membuat kelas yang membungkus operasi TPUOrdinalSelector baru.
Keluaran <Bilangan Bulat>
perangkatOrdinal ()
Vektor 1 atau lebih inti TPU.

Metode Warisan

Metode Publik

Keluaran publik <Bilangan Bulat> sebagai Keluaran ()

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.

pembuatan TPUOrdinalSelector statis publik ( Lingkup lingkup)

Metode pabrik untuk membuat kelas yang membungkus operasi TPUOrdinalSelector baru.

Parameter
cakupan ruang lingkup saat ini
Kembali
  • contoh baru TPUOrdinalSelector

Output publik <Integer> deviceOrdinals ()

Vektor 1 atau lebih inti TPU.