aliran tensor:: operasi:: JarangConcat
#include <sparse_ops.h>
Menggabungkan daftar SparseTensor
sepanjang dimensi yang ditentukan.
Ringkasan
Penggabungan dilakukan sehubungan dengan versi padat dari tensor renggang ini. Diasumsikan bahwa setiap masukan adalah SparseTensor
yang elemen-elemennya diurutkan seiring bertambahnya jumlah dimensi.
Semua bentuk masukan harus cocok, kecuali dimensi gabungan. Daftar indices
, values
, dan shapes
harus memiliki panjang yang sama.
Bentuk keluaran identik dengan masukan, kecuali sepanjang dimensi concat, yang merupakan jumlah dari ukuran masukan sepanjang dimensi tersebut.
Elemen keluaran akan digunakan untuk mempertahankan urutan pengurutan seiring bertambahnya jumlah dimensi.
Operasi ini berjalan dalam waktu O(M log M)
, di mana M
adalah jumlah total nilai tidak kosong di semua input. Hal ini disebabkan oleh kebutuhan akan pengurutan internal agar dapat digabungkan secara efisien di seluruh dimensi yang berubah-ubah.
Misalnya, jika concat_dim = 1
dan inputnya adalah
sp_inputs[0]: shape = [2, 3] [0, 2]: "a" [1, 0]: "b" [1, 1]: "c" sp_inputs[1]: shape = [2, 4] [0, 1]: "d" [0, 2]: "e"
maka outputnya adalah
shape = [2, 7] [0, 2]: "a" [0, 4]: "d" [0, 5]: "e" [1, 0]: "b" [1, 1]: "c"
Secara grafis ini setara dengan melakukan
[ a] concat [ d e ] = [ a d e ] [b c ] [ ] [b c ]
Argumen:
- ruang lingkup: Objek Lingkup
- indeks: 2-D. Indeks setiap masukan
SparseTensor
. - nilai: 1-D. Nilai tidak kosong dari setiap
SparseTensor
. - bentuk: 1-D. Bentuk setiap
SparseTensor
. - concat_dim: Dimensi untuk digabungkan. Harus berada dalam rentang [-rank, rank), dengan rank adalah jumlah dimensi di setiap masukan
SparseTensor
.
Pengembalian:
-
Output
keluaran_indeks: 2-D. Indeks dari gabunganSparseTensor
. - Nilai keluaran
Output
: 1-D. Nilai tak kosong dari gabunganSparseTensor
. - Bentuk keluaran
Output
: 1-D. BentukSparseTensor
yang digabungkan.
Konstruktor dan Destruktor | |
---|---|
SparseConcat (const :: tensorflow::Scope & scope, :: tensorflow::InputList indices, :: tensorflow::InputList values, :: tensorflow::InputList shapes, int64 concat_dim) |
Atribut publik | |
---|---|
operation | |
output_indices | |
output_shape | |
output_values |
Atribut publik
operasi
Operation operation
keluaran_indeks
::tensorflow::Output output_indices
keluaran_bentuk
::tensorflow::Output output_shape
nilai_output
::tensorflow::Output output_values
Fungsi publik
JarangConcat
SparseConcat( const ::tensorflow::Scope & scope, ::tensorflow::InputList indices, ::tensorflow::InputList values, ::tensorflow::InputList shapes, int64 concat_dim )