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 dari SparseTensor , ukuran [nnz, 2] Matriks.
  • a_nilai: 1-D. values dari SparseTensor , 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:

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
)