fluxo tensor:: ops:: SparseMatMul
#include <math_ops.h>Multiplique a matriz “a” pela matriz “b”.
Resumo
As entradas devem ser matrizes bidimensionais e a dimensão interna de “a” deve corresponder à dimensão externa de “b”. Tanto "a" quanto "b" devem ser Tensor s e não SparseTensor s. Esta operação é otimizada para o caso em que pelo menos um de "a" ou "b" é esparso, no sentido de que possuem uma grande proporção de valores zero. O ponto de equilíbrio para usar isso versus uma multiplicação de matriz densa em uma plataforma foi de 30% de valores zero na matriz esparsa.
O cálculo do gradiente desta operação só aproveitará a dispersão no gradiente de entrada quando esse gradiente vier de um Relu .
Argumentos:
- escopo: um objeto Escopo
Retorna:
-
Output: O tensor do produto.
Construtores e Destruidores | |
|---|---|
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) |
Funções públicas | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
Funções estáticas públicas | |
|---|---|
AIsSparse (bool x) | |
BIsSparse (bool x) | |
TransposeA (bool x) | |
TransposeB (bool x) | |
Estruturas | |
|---|---|
| tensorflow:: ops:: SparseMatMul:: Attrs | Configuradores de atributos opcionais para SparseMatMul . |
Atributos públicos
operação
Operation operation
produto
::tensorflow::Output product
Funções públicas
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
)
nó
::tensorflow::Node * node() const