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:

Kehrt zurück:

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
)