InfeedEnqueueTuple

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
buat ( Lingkup lingkup, input Iterable< Operand <?>>, Daftar< Bentuk > bentuk, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi InfeedEnqueueTuple baru.
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 input "bentuk". Elemen tata letak untuk sub-bentuk dapat diatur ke -1, dalam hal ini tata letak terkait akan dihitung dengan operasi pemasukan.