aliran tensor:: operasi:: SparseTensorDenseMatMul
#include <sparse_ops.h>
Kalikan SparseTensor (peringkat 2) "A" dengan matriks padat "B".
Ringkasan
Tidak ada pemeriksaan validitas yang dilakukan pada indeks A. Namun, format input berikut disarankan untuk perilaku optimal:
if adjoint_a == false: A harus diurutkan dalam urutan leksikografis yang meningkat. Gunakan SparseReorder jika Anda tidak yakin. if adjoint_a == true: A harus diurutkan berdasarkan peningkatan dimensi 1 (yaitu, urutan "kolom utama" dan bukan urutan "baris utama").
Argumen:
- ruang lingkup: Objek Lingkup
- a_indeks: 2-D.
indices
dariSparseTensor
, ukuran[nnz, 2]
Matriks. - a_nilai: 1-D.
values
dariSparseTensor
, ukuran[nnz]
Vektor. - a_bentuk: 1-D.
shape
SparseTensor
, ukuran[2]
Vektor. - b: 2-D. Matriks yang padat.
Atribut opsional (lihat Attrs
):
- adjoint_a: Gunakan adjoint A dalam perkalian matriks. Jika A kompleks, maka transpose(conj(A)). Kalau tidak, itu transpose(A).
- adjoint_b: Gunakan adjoint B dalam perkalian matriks. Jika B kompleks, ini adalah transpose(conj(B)). Kalau tidak, itu transpose(B).
Pengembalian:
-
Output
: Tensor produk.
Konstruktor dan Destruktor | |
---|---|
SparseTensorDenseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a_indices, :: tensorflow::Input a_values, :: tensorflow::Input a_shape, :: tensorflow::Input b) | |
SparseTensorDenseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a_indices, :: tensorflow::Input a_values, :: tensorflow::Input a_shape, :: tensorflow::Input b, const SparseTensorDenseMatMul::Attrs & attrs) |
Atribut publik | |
---|---|
operation | |
product |
Fungsi publik | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Fungsi statis publik | |
---|---|
AdjointA (bool x) | |
AdjointB (bool x) |
Struktur | |
---|---|
tensorflow:: ops:: SparseTensorDenseMatMul:: Attrs | Penyetel atribut opsional untuk SparseTensorDenseMatMul . |
Atribut publik
operasi
Operation operation
produk
::tensorflow::Output product
Fungsi publik
SparseTensorDenseMatMul
SparseTensorDenseMatMul( const ::tensorflow::Scope & scope, ::tensorflow::Input a_indices, ::tensorflow::Input a_values, ::tensorflow::Input a_shape, ::tensorflow::Input b )
SparseTensorDenseMatMul
SparseTensorDenseMatMul( const ::tensorflow::Scope & scope, ::tensorflow::Input a_indices, ::tensorflow::Input a_values, ::tensorflow::Input a_shape, ::tensorflow::Input b, const SparseTensorDenseMatMul::Attrs & attrs )
simpul
::tensorflow::Node * node() const
operator::tensorflow::Masukan
operator::tensorflow::Input() const
operator::tensorflow::Keluaran
operator::tensorflow::Output() const
Fungsi statis publik
BerdampinganA
Attrs AdjointA( bool x )
BerdampinganB
Attrs AdjointB( bool x )