aliran tensor:: operasi:: JarangMatMul

#include <math_ops.h>

Kalikan matriks "a" dengan matriks "b".

Ringkasan

Inputnya harus berupa matriks dua dimensi dan dimensi dalam "a" harus sesuai dengan dimensi luar "b". Baik "a" maupun "b" harus berupa Tensor , bukan SparseTensor . Operasi ini dioptimalkan untuk kasus di mana setidaknya salah satu dari "a" atau "b" jarang, dalam arti bahwa keduanya memiliki sebagian besar nilai nol. Titik impas untuk menggunakan ini versus perkalian matriks padat pada satu platform adalah 30% nilai nol dalam matriks renggang.

Komputasi gradien pada operasi ini hanya akan memanfaatkan ketersebaran pada gradien masukan jika gradien tersebut berasal dari Relu .

Argumen:

Pengembalian:

Konstruktor dan Destruktor

SparseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b)
SparseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b, const SparseMatMul::Attrs & attrs)

Atribut publik

operation
product

Fungsi publik

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Fungsi statis publik

AIsSparse (bool x)
BIsSparse (bool x)
TransposeA (bool x)
TransposeB (bool x)

Struktur

tensorflow:: ops:: SparseMatMul:: Attrs

Penyetel atribut opsional untuk SparseMatMul .

Atribut publik

operasi

Operation operation

produk

::tensorflow::Output product

Fungsi publik

JarangMatMul

 SparseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b
)

JarangMatMul

 SparseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b,
  const SparseMatMul::Attrs & attrs
)

simpul

::tensorflow::Node * node() const 

operator::tensorflow::Masukan

 operator::tensorflow::Input() const 

operator::tensorflow::Keluaran

 operator::tensorflow::Output() const 

Fungsi statis publik

AIsSparse

Attrs AIsSparse(
  bool x
)

BIsSparse

Attrs BIsSparse(
  bool x
)

Mengubah urutanA

Attrs TransposeA(
  bool x
)

TransposB

Attrs TransposeB(
  bool x
)