tensorflow::ops::SparseMatMul

Multiplique a matriz "a" pela matriz "b".

Resumo

As entradas devem ser matrizes bidimensionais e a dimensão interna de "a" deve corresponder à dimensão externa de "b". Tanto "a" quanto "b" devem ser Tensor , não SparseTensor . Esta operação é otimizada para o caso em que pelo menos uma das matrizes "a" ou "b" seja esparsa, no sentido de que possua uma grande proporção de valores zero. O ponto de equilíbrio entre usar esta operação e uma multiplicação de matrizes densas em uma plataforma foi de 30% de valores zero na matriz esparsa.

O cálculo do gradiente desta operação só aproveitará a esparsidade no gradiente de entrada quando esse gradiente vier de uma função ReLU .

Argumentos:

• escopo: Um objeto de escopo

Devoluções:

• Output : O tensor produto.

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
)