Tensorflow :: ops :: SparseMatMul
#include <math_ops.h>
Multiplizieren Sie die Matrix "a" mit der Matrix "b".
Zusammenfassung
Die Eingaben müssen zweidimensionale Matrizen sein und die innere Dimension von "a" muss mit der äußeren Dimension von "b" übereinstimmen. Sowohl "a" und "b" muss Tensor
s nicht SparseTensor
s. Diese Operation ist für den Fall optimiert, dass mindestens eines von "a" oder "b" in dem Sinne spärlich ist, dass sie einen großen Anteil von Nullwerten haben. Die Gewinnschwelle für die Verwendung dieser gegenüber einer dichten Matrixmultiplikation auf einer Plattform betrug 30% Nullwerte in der dünn besetzten Matrix.
Die Gradientenberechnung dieser Operation nutzt die Sparsity im Eingabegradienten nur dann aus, wenn dieser Gradient von einem Relu stammt .
Argumente:
- scope: Ein Scope- Objekt
Kehrt zurück:
-
Output
: Der Produkttensor.
Konstruktoren und Destruktoren | |
---|---|
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) |
Öffentliche Attribute | |
---|---|
operation | |
product |
Öffentliche Funktionen | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Öffentliche statische Funktionen | |
---|---|
AIsSparse (bool x) | |
BIsSparse (bool x) | |
TransposeA (bool x) | |
TransposeB (bool x) |
Strukturen | |
---|---|
tensorflow :: ops :: SparseMatMul :: Attrs | Optionale Attributsetzer für SparseMatMul . |
Öffentliche Attribute
Operation
Operation operation
Produkt
::tensorflow::Output product
Öffentliche Funktionen
SparseMatMul
SparseMatMul( const ::tensorflow::Scope & scope, ::tensorflow::Input a, ::tensorflow::Input b )
SparseMatMul
SparseMatMul( const ::tensorflow::Scope & scope, ::tensorflow::Input a, ::tensorflow::Input b, const SparseMatMul::Attrs & attrs )
Knoten
::tensorflow::Node * node() const
operator :: tensorflow :: Input
operator::tensorflow::Input() const
operator :: tensorflow :: Output
operator::tensorflow::Output() const
Öffentliche statische Funktionen
AIsSparse
Attrs AIsSparse( bool x )
BIsSparse
Attrs BIsSparse( bool x )
TransposeA
Attrs TransposeA( bool x )
TransponierenB
Attrs TransposeB( bool x )