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 )