flujo tensor:: operaciones:: SparseMatMul

#include <math_ops.h>

Multiplicar la matriz "a" por la matriz "b".

Resumen

Las entradas deben ser matrices bidimensionales y la dimensión interior de "a" debe coincidir con la dimensión exterior de "b". Tanto "a" como "b" deben ser Tensor s, no SparseTensor s. Esta operación está optimizada para el caso en el que 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 al usar esto versus una matriz densa multiplicada en una plataforma fue del 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:

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)

Atributos públicos

operation
product

Funciones públicas

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:: operaciones:: SparseMatMul:: Atributos

Configuradores de atributos opcionales para SparseMatMul .

Atributos públicos

operación

Operation operation

producto

::tensorflow::Output product

Funciones 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
)

nodo

::tensorflow::Node * node() const 

operador::tensorflow::Entrada

 operator::tensorflow::Input() const 

operador::tensorflow::Salida

 operator::tensorflow::Output() const 

Funciones estáticas públicas

AI es escasa

Attrs AIsSparse(
  bool x
)

BIsEscaso

Attrs BIsSparse(
  bool x
)

TransponerA

Attrs TransposeA(
  bool x
)

Transponer B

Attrs TransposeB(
  bool x
)