tensorflow::ops::SparseMatMul
#include <math_ops.h>
Multiply матрицу «a» на матрицу «b».
Краткое описание
Входные данные должны быть двумерными матрицами, а внутренний размер «a» должен совпадать с внешним размером «b». И «a», и «b» должны быть Tensors, а не SparseTensors. Эта операция оптимизирована для случаев, когда хотя бы одна из матриц «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 operation
product
::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 )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
Публичные статические функции
AIsSparse
Attrs AIsSparse( bool x )
BIsSparse
Attrs BIsSparse( bool x )
TransposeA
Attrs TransposeA( bool x )
TransposeB
Attrs TransposeB( bool x )
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/sparse-mat-mul