Peringatan: API ini tidak digunakan lagi dan akan dihapus di versi TensorFlow mendatang setelah penggantinya stabil.

BarrierTakeMany

public class akhir BarrierTakeMany

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).

publik Keluaran <String> kunci ()

Tensor kunci satu dimensi, dengan panjang num_elements.

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.

public static BarrierTakeMany.Options waitForIncomplete (Boolean waitForIncomplete)