tensoreflusso:: ops:: SparseMatMul

Moltiplicare la matrice "a" per la matrice "b".

Riepilogo

Gli input devono essere matrici bidimensionali e la dimensione interna di "a" deve corrispondere alla dimensione esterna di "b". Sia "a" che "b" devono essere Tensor s e non SparseTensor s. Questa operazione è ottimizzata per il caso in cui almeno uno tra "a" e "b" è sparso, nel senso che hanno una grande proporzione di valori zero. Il pareggio per l'utilizzo di questo rispetto a una matrice densa moltiplicata su una piattaforma era pari al 30% di valori zero nella matrice sparsa.

Il calcolo del gradiente di questa operazione trarrà vantaggio dalla scarsità nel gradiente di input solo quando tale gradiente proviene da un Relu .

Argomenti:

• scope: un oggetto Scope

Ritorna:

• Output : il tensore del prodotto.

Attributi pubblici

Operation operation

::tensorflow::Output product

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs AIsSparse(
  bool x
)

Attrs BIsSparse(
  bool x
)

Attrs TransposeA(
  bool x
)

Attrs TransposeB(
  bool x
)