tensorflow::ops::BarrierTakeMany

Mengambil sejumlah elemen yang telah selesai dari sebuah penghalang.

Ringkasan

Operasi ini menggabungkan tensor komponen elemen lengkap di sepanjang dimensi ke-0 untuk membuat tensor komponen tunggal.

Elemen-elemen keluar dari penghalang ketika sudah lengkap, dan sesuai urutan saat dimasukkan ke dalam penghalang. Keluaran indeks memberikan informasi tentang kelompok (batch) di mana setiap elemen awalnya dimasukkan ke dalam penghalang.

Argumen:

• Cakupan: Objek Cakupan
• pegangan: Pegangan pada sebuah penghalang.
• num_elements: Tensor elemen tunggal yang berisi jumlah elemen yang akan diambil.
• component_types: Tipe setiap komponen dalam suatu nilai.

Atribut opsional (lihat Attrs ):

• allow_small_batch: Mengizinkan pengembalian item kurang dari num_elements jika barrier sudah tertutup.
• timeout_ms: Jika antrean kosong, operasi ini akan diblokir hingga timeout_ms milidetik. Catatan: Opsi ini belum didukung.

Mengembalikan:

• Indeks Output : Tensor satu dimensi berisi indeks, dengan panjang num_elems. Indeks ini mengacu pada batch di mana nilai-nilai ditempatkan ke dalam barrier (dimulai dengan MIN_LONG dan meningkat dengan setiap BarrierInsertMany ).
• Output keys: Sebuah tensor satu dimensi berisi kunci, dengan panjang num_elements.
• Nilai OutputList : Satu tensor berdimensi sembarang per komponen dalam elemen penghalang. Semua nilai memiliki panjang num_elements pada dimensi ke-0.

Atribut publik

::tensorflow::Output indices

::tensorflow::Output keys

Operation operation

::tensorflow::OutputList values

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types
)

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types,
  const BarrierTakeMany::Attrs & attrs
)

Attrs AllowSmallBatch(
  bool x
)

Attrs TimeoutMs(
  int64 x
)

Attrs WaitForIncomplete(
  bool x
)