Mengambil sejumlah elemen lengkap dari penghalang.
Operasi ini menggabungkan tensor komponen elemen yang telah selesai sepanjang dimensi ke-0 untuk membuat tensor komponen tunggal.
Elemen keluar dari penghalang ketika sudah selesai, dan sesuai urutan penempatannya 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
BarrierTakeMany.Options statis | izinkanSmallBatch (Boolean izinkanSmallBatch) |
Penghalang statisTakeMany | |
Keluaran <Panjang> | indeks () Tensor indeks satu dimensi, dengan panjang num_elems. |
Keluaran <String> | kunci () Tensor kunci satu dimensi, dengan panjang num_elements. |
BarrierTakeMany.Options statis | timeoutMs (Waktu tunggu yang lama) |
Daftar< Keluaran <?>> | nilai () Satu tensor berdimensi apa pun per komponen dalam elemen penghalang. |
BarrierTakeMany.Options statis | waitForIncomplete (Boolean waitForIncomplete) |
Metode Warisan
Metode Publik
BarrierTakeMany.Options statis publik memungkinkanSmallBatch (Boolean memungkinkanSmallBatch)
Parameter
izinkan SmallBatch | Izinkan untuk mengembalikan item kurang dari num_elements jika penghalang sudah ditutup. |
---|
public static BarrierTakeMany buat ( Lingkup cakupan , pegangan Operan <String>, Operan <Integer> numElements, Daftar<Kelas<?>> Tipe komponen, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi BarrierTakeMany baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
menangani | Pegangan ke penghalang. |
numElements | 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
Indeks Output publik <Panjang> ()
Tensor indeks satu dimensi, dengan panjang num_elems. Indeks ini mengacu pada kumpulan nilai yang ditempatkan ke dalam penghalang (dimulai dengan MIN_LONG dan meningkat setiap BarrierInsertMany).
public static BarrierTakeMany.Options timeoutMs (Waktu tunggu lama)
Parameter
batas waktuMs | Jika antrean kosong, operasi ini akan diblokir hingga timeout_ms milidetik. Catatan: Opsi ini belum didukung. |
---|
Daftar publik< Keluaran <?>> nilai ()
Satu tensor berdimensi apa pun per komponen dalam elemen penghalang. Semua nilai memiliki panjang num_elements di dimensi ke-0.