aliran tensor:: operasi:: PenghalangAmbilBanyak

Mengambil sejumlah elemen lengkap dari penghalang.

Ringkasan

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.

Argumen:

• ruang lingkup: Objek Lingkup
• menangani: Pegangan untuk penghalang.
• num_elements: Tensor elemen tunggal yang berisi jumlah elemen yang akan diambil.
• komponen_tipe: Jenis setiap komponen dalam suatu nilai.

Atribut opsional (lihat Attrs ):

• izinkan_kecil_batch: Izinkan untuk mengembalikan kurang dari num_elements item jika penghalang sudah ditutup.
• timeout_ms: Jika antrian kosong, operasi ini akan diblokir hingga timeout_ms milidetik. Catatan: Opsi ini belum didukung.

Pengembalian:

• Indeks Output : 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 Output : Tensor kunci satu dimensi, dengan panjang num_elements.
• Nilai OutputList : Satu tensor dimensi apa pun per komponen dalam elemen penghalang. Semua nilai memiliki panjang num_elements di 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
)