tensorflow :: operaciones :: SparseMatMul

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 provenga de un Relu .

Argumentos:

• alcance: un objeto de alcance

Devoluciones:

• Output : el tensor del producto.

Atributos públicos

Operation operation

::tensorflow::Output product

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs AIsSparse(
  bool x
)

Attrs BIsSparse(
  bool x
)

Attrs TransposeA(
  bool x
)

Attrs TransposeB(
  bool x
)