public class akhir InfeedEnqueueTuple
Mengumpankan beberapa nilai Tensor ke dalam komputasi sebagai tupel XLA.
Kelas Bersarang
kelas | InfeedEnqueueTuple.Options | Atribut opsional untuk InfeedEnqueueTuple |
Metode Publik
statis InfeedEnqueueTuple | membuat ( Lingkup lingkup, Iterable < Operan <? >> masukan, Daftar < Shape > bentuk, Options ... pilihan) Metode pabrik untuk membuat kelas yang membungkus operasi InfeedEnqueueTuple baru. |
statis InfeedEnqueueTuple.Options | deviceOrdinal (Long deviceOrdinal) |
statis InfeedEnqueueTuple.Options | layout (Daftar <Panjang> layout) |
Metode yang Diwarisi
Metode Publik
public static InfeedEnqueueTuple membuat ( Scope lingkup, Iterable < Operan <? >> masukan, Daftar < Shape > bentuk, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi InfeedEnqueueTuple baru.
Parameter
cakupan | lingkup saat ini |
---|---|
masukan | Daftar tensor yang akan diberikan menggunakan mekanisme infeed. |
bentuk | Bentuk setiap tensor dalam `input`. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari InfeedEnqueueTuple
public static InfeedEnqueueTuple.Options deviceOrdinal (Long deviceOrdinal)
Parameter
perangkatOrdinal | Perangkat TPU yang akan digunakan. Ini harus -1 ketika Op berjalan pada perangkat TPU, dan >= 0 saat Op berjalan pada perangkat CPU. |
---|
public static InfeedEnqueueTuple.Options layout (Daftar <Panjang> layout)
Parameter
tata letak | Sebuah vektor memegang tata letak yang diminta dalam urutan minor-ke-mayor untuk semua bentuk tupel, dalam urutan bentuk muncul di input "bentuk". Elemen tata letak untuk sub-bentuk dapat diatur ke -1, dalam hal ini tata letak yang sesuai akan dihitung oleh operasi pengumpanan. |
---|