flujo tensor:: operaciones:: SparseMatMul

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 que al menos uno de "a" o "b" sea 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:

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