tensorflow:: ops:: SparseMatMul

Multiply matrix "a" by matrix "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 Tensor , no SparseTensor . Esta operación está optimizada para el caso en que al menos una de las matrices, "a" o "b", sea dispersa, es decir, tenga una gran proporción de valores cero. El punto de equilibrio al usar esta operación en comparación con una multiplicación de matrices densas 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 una Relu .

Argumentos:

• ámbito: Un objeto de ámbito

Devoluciones:

• Output : El tensor 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
)