BarrierTakeMany

kelas akhir publik BarrierTakeMany

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
buat ( Lingkup cakupan , pegangan Operan <String>, Operan <Bilangan Bulat> numElements, Daftar<Kelas<?>> Tipe komponen, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi BarrierTakeMany baru.
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).

Kunci <String> Output publik ()

Tensor kunci satu dimensi, dengan panjang num_elements.

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.

BarrierTakeMany.Options waitForIncomplete publik statis (Boolean waitForIncomplete)