fluxo tensor:: 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 s e não SparseTensor s. Esta operação é otimizada para o caso em que pelo menos um de "a" ou "b" é esparso, no sentido de que possuem uma grande proporção de valores zero. O ponto de equilíbrio para usar isso versus uma multiplicação de matriz densa em uma plataforma foi de 30% de valores zero na matriz esparsa.

O cálculo do gradiente desta operação só aproveitará a dispersão no gradiente de entrada quando esse gradiente vier de um Relu .

Argumentos:

• escopo: um objeto Escopo

Retorna:

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