tensorflow :: ops :: MatMul

bookmark_border Mantenha tudo organizado com as coleções Salve e categorize o conteúdo com base nas suas preferências.

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

Resumo

As entradas devem ser matrizes bidimensionais e a dimensão interna de "a" (após ser transposta se transpose_a for true) deve corresponder à dimensão externa de "b" (após ser transposta se transposed_b for true).

Nota : A implementação de kernel padrão para MatMul em GPUs usa cublas.

Argumentos:

• escopo: um objeto Scope

Atributos opcionais (consulte Attrs ):

• transpose_a: se verdadeiro, "a" é transposto antes da multiplicação.
• transpose_b: se verdadeiro, "b" é transposto antes da multiplicação.

Retorna:

• Output : O tensor do produto.

Atributos públicos

Operation operation

::tensorflow::Output product

 MatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b
)

 MatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b,
  const MatMul::Attrs & attrs
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs TransposeA(
  bool x
)

Attrs TransposeB(
  bool x
)