aliran tensor:: operasi:: PenghalangAmbil Banyak

#include <data_flow_ops.h>

Mengambil jumlah elemen selesai yang diberikan dari penghalang.

Ringkasan

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.

Argumen:

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

Atribut opsional (lihat Attrs ):

  • allow_small_batch: Memungkinkan untuk mengembalikan kurang dari num_element item jika penghalang sudah ditutup.
  • timeout_ms: Jika antrian kosong, operasi ini akan memblokir hingga timeout_ms milidetik. Catatan: Opsi ini belum didukung.

Pengembalian:

  • Indeks Output : Tensor satu dimensi dari indeks, 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 ).
  • 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 num_elements panjang dalam dimensi 0.

Konstruktor dan Destructor

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

PenghalangAmbil Banyak

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

PenghalangAmbil Banyak

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

Fungsi statis publik

IzinkanSmallBatch

Attrs AllowSmallBatch(
  bool x
)

Waktu habisMs

Attrs TimeoutMs(
  int64 x
)

Tunggu Belum Selesai

Attrs WaitForIncomplete(
  bool x
)