Mengambil jumlah elemen selesai yang diberikan dari penghalang.
Operasi ini menggabungkan tensor komponen elemen lengkap di sepanjang dimensi ke-0 untuk membuat tensor komponen tunggal.
Elemen keluar dari penghalang ketika mereka selesai, dan dalam urutan di mana mereka ditempatkan ke dalam penghalang. Output indeks memberikan informasi tentang batch di mana setiap elemen awalnya dimasukkan ke dalam penghalang.
Kelas Bersarang
kelas | BarrierTakeMany.Options | Atribut opsional untuk BarrierTakeMany |
Metode Publik
statis BarrierTakeMany.Options | allowSmallBatch (Boolean allowSmallBatch) |
statis BarrierTakeMany | membuat ( Lingkup lingkup, Operan <String> handle, Operan <Integer> numElements, Daftar <Kelas <? >> componentTypes, Options ... pilihan) Metode pabrik untuk membuat kelas yang membungkus operasi BarrierTakeMany baru. |
Output <Panjang> | indeks () Tensor indeks satu dimensi, dengan panjang num_elems. |
Output <String> | Kunci () Tensor kunci satu dimensi, dengan panjang num_elements. |
statis BarrierTakeMany.Options | timeoutMs (Long timeoutMs) |
Daftar < Keluaran <? >> | nilai-nilai () Satu tensor dimensi apa pun per komponen dalam elemen penghalang. |
statis BarrierTakeMany.Options | waitForIncomplete (Boolean waitForIncomplete) |
Metode yang Diwarisi
Metode Publik
public static BarrierTakeMany.Options allowSmallBatch (Boolean allowSmallBatch)
Parameter
izinkanSmallBatch | Izinkan untuk mengembalikan kurang dari num_element item jika penghalang sudah ditutup. |
---|
public static BarrierTakeMany membuat ( Scope lingkup, Operan <String> handle, Operan <Integer> numElements, Daftar <Kelas <? >> componentTypes, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi BarrierTakeMany baru.
Parameter
cakupan | lingkup saat ini |
---|---|
menangani | Pegangan ke penghalang. |
jumlahElemen | Tensor elemen tunggal yang berisi jumlah elemen yang akan diambil. |
tipe komponen | Jenis setiap komponen dalam suatu nilai. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari BarrierTakeMany
publik Keluaran <Panjang> indeks ()
Tensor indeks satu dimensi, dengan panjang num_elems. Indeks ini mengacu pada kumpulan di mana nilai ditempatkan ke dalam penghalang (dimulai dengan MIN_LONG dan meningkat dengan setiap BarrierInsertMany).
public static BarrierTakeMany.Options timeoutMs (Long timeoutMs)
Parameter
waktu habisMs | Jika antrian kosong, operasi ini akan memblokir hingga timeout_ms milidetik. Catatan: Opsi ini belum didukung. |
---|
public List < Keluaran <? >> nilai-nilai ()
Satu tensor dimensi apa pun per komponen dalam elemen penghalang. Semua nilai memiliki num_elements panjang dalam dimensi 0.