InfeedEnqueue kelas akhir publik
Sebuah operasi yang memasukkan satu nilai Tensor ke dalam komputasi.
Kelas Bersarang
kelas | InfeedEnqueue.Options | Atribut opsional untuk InfeedEnqueue |
Metode Publik
statis <T> InfeedEnqueue | |
InfeedEnqueue.Options statis | deviceOrdinal (Perangkat panjangOrdinal) |
InfeedEnqueue.Options statis | tata letak (Daftar<Panjang> tata letak) |
InfeedEnqueue.Options statis |
Metode Warisan
Metode Publik
pembuatan InfeedEnqueue statis publik ( Lingkup lingkup, input Operan <T>, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi InfeedEnqueue baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
memasukkan | Tensor yang akan diberikan menggunakan mekanisme infeed. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari InfeedEnqueue
InfeedEnqueue.Options deviceOrdinal statis publik (PerangkatOrdinal panjang)
Parameter
perangkatOrdinal | Perangkat TPU yang akan digunakan. Nilainya harus -1 saat Operasi dijalankan pada perangkat TPU, dan >= 0 saat Operasi dijalankan pada perangkat CPU. |
---|
tata letak InfeedEnqueue.Options statis publik (tata letak Daftar<Panjang>)
Parameter
tata letak | Sebuah vektor yang menampung tata letak yang diminta dalam urutan minor-ke-mayor. Jika atribut layout diteruskan, namun nilainya semuanya -1, layout akan dihitung dengan operasi infeed. |
---|