tensorflow::ops::SparseMatMul
#include <math_ops.h>Умножьте матрицу "a" на матрицу "b".
Краткое содержание
Входные данные должны представлять собой двумерные матрицы, при этом внутренняя размерность матрицы "a" должна совпадать с внешней размерностью матрицы "b". Обе матрицы "a" и "b" должны быть Tensor , а не SparseTensor тензорами. Эта операция оптимизирована для случая, когда хотя бы одна из матриц "a" или "b" является разреженной в том смысле, что в ней большая доля нулевых значений. Точка безубыточности при использовании этого метода по сравнению с умножением на плотную матрицу на одной из платформ составила 30% нулевых значений в разреженной матрице.
При вычислении градиента в этой операции преимущество разреженности входного градиента будет использоваться только в том случае, если этот градиент получен с помощью функции Relu .
Аргументы:
- scope: Объект Scope
Возвращает:
-
Output: тензор произведения.
Конструкторы и Деструкторы | |
|---|---|
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) |
Общедоступные атрибуты | |
|---|---|
operation | |
product | |
Общественные функции | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
Открытые статические функции | |
|---|---|
AIsSparse (bool x) | |
BIsSparse (bool x) | |
TransposeA (bool x) | |
TransposeB (bool x) | |
Структуры | |
|---|---|
| tensorflow::ops::SparseMatMul::Attrs | Дополнительные параметры установки атрибутов для SparseMatMul . |
Общедоступные атрибуты
операция
Operation operation
продукт
::tensorflow::Output product
Общественные функции
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 )
узел
::tensorflow::Node * node() const
оператор::tensorflow::Input
operator::tensorflow::Input() const
оператор::tensorflow::Output
operator::tensorflow::Output() const
Открытые статические функции
AIsSparse
Attrs AIsSparse( bool x )
BIsSparse
Attrs BIsSparse( bool x )
ТранспонироватьA
Attrs TransposeA( bool x )
ТранспонированиеB
Attrs TransposeB( bool x )