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 )
узел
::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 )
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/versions/r2.9/api_docs/cc/class/tensorflow/ops/sparse-mat-mul