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