kelas statis publik InfeedEnqueueTuple.Options
Atribut opsional untuk InfeedEnqueueTuple
Metode Publik
InfeedEnqueueTuple.Options | deviceOrdinal (Perangkat panjangOrdinal) |
InfeedEnqueueTuple.Options | tata letak (Daftar<Panjang> tata letak) |
Metode Warisan
Metode Publik
public InfeedEnqueueTuple.Options deviceOrdinal (Perangkat PanjangOrdinal)
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 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. |
---|