kelas akhir publik InfeedEnqueueTuple
Memasukkan beberapa nilai Tensor ke dalam komputasi sebagai tupel XLA.
Kelas Bersarang
kelas | InfeedEnqueueTuple.Options | Atribut opsional untuk InfeedEnqueueTuple |
Metode Publik
InfeedEnqueueTuple statis | |
InfeedEnqueueTuple.Options statis | deviceOrdinal (Perangkat panjangOrdinal) |
InfeedEnqueueTuple.Options statis | tata letak (Daftar<Panjang> tata letak) |
Metode Warisan
Metode Publik
public static InfeedEnqueueTuple buat ( Lingkup lingkup, input Iterable< Operand <?>>, Daftar< Bentuk > bentuk, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi InfeedEnqueueTuple baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
masukan | Daftar tensor yang akan diberikan menggunakan mekanisme infeed. |
bentuk | Bentuk setiap tensor di `input`. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari InfeedEnqueueTuple
public static InfeedEnqueueTuple.Options deviceOrdinal (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 InfeedEnqueueTuple.Options statis publik (tata letak Daftar<Panjang>)
Parameter
tata letak | Sebuah vektor yang menampung tata letak yang diminta dalam urutan minor hingga mayor untuk semua bentuk tuple, sesuai urutan kemunculan bentuk di masukan "bentuk". Elemen tata letak untuk sub-bentuk dapat diatur ke -1, dalam hal ini tata letak terkait akan dihitung dengan operasi pemasukan. |
---|