aliran tensor:: operasi:: JarangTambahkan
#include <sparse_ops.h>
Menambahkan dua objek SparseTensor
untuk menghasilkan SparseTensor
lainnya.
Ringkasan
Indeks objek input SparseTensor
diasumsikan diurutkan dalam urutan leksikografis standar. Jika tidak demikian, sebelum langkah ini jalankan SparseReorder
untuk memulihkan pengurutan indeks.
Secara default, jika dua nilai berjumlah nol pada suatu indeks, output SparseTensor
akan tetap menyertakan lokasi tertentu tersebut dalam indeksnya, menyimpan angka nol di slot nilai yang sesuai. Untuk mengesampingkan hal ini, penelepon dapat menentukan thresh
, yang menunjukkan bahwa jika jumlah tersebut memiliki besaran yang lebih kecil dari thresh
, maka nilai dan indeks terkait tidak akan disertakan. Secara khusus, thresh == 0
(default) berarti semuanya dipertahankan dan ambang batas aktual hanya terjadi untuk nilai positif.
Dalam bentuk berikut, nnz
adalah hitungan setelah memperhitungkan thresh
.
Argumen:
- ruang lingkup: Objek Lingkup
- a_indeks: 2-D.
indices
dariSparseTensor
pertama, ukuran Matriks[nnz, ndims]
. - a_nilai: 1-D.
values
SparseTensor
pertama, ukuran[nnz]
Vektor. - a_bentuk: 1-D.
shape
SparseTensor
pertama, ukuran[ndims]
Vektor. - b_indeks: 2-D.
indices
dariSparseTensor
kedua, ukuran Matriks[nnz, ndims]
. - b_nilai: 1-D.
values
SparseTensor
kedua, ukuran[nnz]
Vektor. - b_bentuk: 1-D.
shape
SparseTensor
kedua, ukuran[ndims]
Vektor. - mengirik: 0-D. Ambang batas besaran yang menentukan apakah nilai keluaran/pasangan indeks memerlukan ruang.
Pengembalian:
Konstruktor dan Destruktor | |
---|---|
SparseAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input a_indices, :: tensorflow::Input a_values, :: tensorflow::Input a_shape, :: tensorflow::Input b_indices, :: tensorflow::Input b_values, :: tensorflow::Input b_shape, :: tensorflow::Input thresh) |
Atribut publik | |
---|---|
operation | |
sum_indices | |
sum_shape | |
sum_values |
Atribut publik
operasi
Operation operation
jumlah_indeks
::tensorflow::Output sum_indices
jumlah_bentuk
::tensorflow::Output sum_shape
jumlah_nilai
::tensorflow::Output sum_values
Fungsi publik
JarangTambahkan
SparseAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input a_indices, ::tensorflow::Input a_values, ::tensorflow::Input a_shape, ::tensorflow::Input b_indices, ::tensorflow::Input b_values, ::tensorflow::Input b_shape, ::tensorflow::Input thresh )