aliran tensor:: operasi:: PenghalangAmbilBanyak

#include <data_flow_ops.h>

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-indeks ini mengacu pada kelompok di mana nilai-nilai 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.

Konstruktor dan Destruktor

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)

Atribut publik

indices
keys
operation
values

Fungsi statis publik

AllowSmallBatch (bool x)
TimeoutMs (int64 x)
WaitForIncomplete (bool x)

Struktur

tensorflow:: ops:: BarrierTakeMany:: Attrs

Penyetel atribut opsional untuk BarrierTakeMany .

Atribut publik

indeks

::tensorflow::Output indices

kunci

::tensorflow::Output keys

operasi

Operation operation

nilai-nilai

::tensorflow::OutputList values

Fungsi publik

PenghalangAmbilBanyak

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

PenghalangAmbilBanyak

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

Fungsi statis publik

Izinkan SmallBatch

Attrs AllowSmallBatch(
  bool x
)

Batas waktuMs

Attrs TimeoutMs(
  int64 x
)

TungguTidak Lengkap

Attrs WaitForIncomplete(
  bool x
)