tensorflow :: operaciones :: SparseMatMul
#include <math_ops.h>Multiplique la matriz "a" por la matriz "b".
Resumen
Las entradas deben ser matrices bidimensionales y la dimensión interna de "a" debe coincidir con la dimensión externa de "b". Tanto "a" como "b" deben ser de Tensor , no de SparseTensor . Esta operación está optimizada para el caso donde al menos uno de "a" o "b" es escaso, en el sentido de que tienen una gran proporción de valores cero. El punto de equilibrio para usar esto versus una matriz densa multiplicada en una plataforma fue 30% de valores cero en la matriz dispersa.
El cálculo del gradiente de esta operación solo aprovechará la escasez en el gradiente de entrada cuando ese gradiente provenga de un Relu .
Argumentos:
- alcance: un objeto de alcance
Devoluciones:
-
Output: el tensor del producto.
Constructores y Destructores | |
|---|---|
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) |
Funciones publicas | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
Funciones estáticas públicas | |
|---|---|
AIsSparse (bool x) | |
BIsSparse (bool x) | |
TransposeA (bool x) | |
TransposeB (bool x) | |
Estructuras | |
|---|---|
| tensorflow :: ops :: SparseMatMul :: Attrs | Establecedores de atributos opcionales para SparseMatMul . |
Atributos públicos
operación
Operation operation
producto
::tensorflow::Output product
Funciones publicas
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
)
nodo
::tensorflow::Node * node() const
operador :: tensorflow :: Entrada
operator::tensorflow::Input() const