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 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 para usar esto frente a 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 proviene 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 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 

operador :: tensorflow :: Salida

 operator::tensorflow::Output() const 

Funciones estáticas públicas

AIsSparse

Attrs AIsSparse(
  bool x
)

BIsSparse

Attrs BIsSparse(
  bool x
)

TransponerA

Attrs TransposeA(
  bool x
)

TransponerB

Attrs TransposeB(
  bool x
)