tensoreflusso:: ops:: MatMul

#include <math_ops.h>

Moltiplicare la matrice "a" per la matrice "b".

Riepilogo

Gli input devono essere matrici bidimensionali e la dimensione interna di "a" (dopo essere stata trasposta se transpose_a è vera) deve corrispondere alla dimensione esterna di "b" (dopo essere stata trasposta se transposed_b è vero).

Nota : l'implementazione predefinita del kernel per MatMul sulle GPU utilizza cublas.

Argomenti:

Attributi facoltativi (vedi Attrs ):

  • transpose_a: se vero, "a" viene trasposto prima della moltiplicazione.
  • transpose_b: se vero, "b" viene trasposto prima della moltiplicazione.

Ritorna:

  • Output : il tensore del prodotto.

Costruttori e distruttori

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)

Attributi pubblici

operation
product

Funzioni pubbliche

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Funzioni pubbliche statiche

TransposeA (bool x)
TransposeB (bool x)

Strutture

tensorflow:: ops:: MatMul:: Attrs

Setter di attributi facoltativi per MatMul .

Attributi pubblici

operazione

Operation operation

Prodotto

::tensorflow::Output product

Funzioni pubbliche

MatMul

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

MatMul

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

nodo

::tensorflow::Node * node() const 

operatore::tensorflow::Input

 operator::tensorflow::Input() const 

operatore::tensorflow::Output

 operator::tensorflow::Output() const 

Funzioni pubbliche statiche

TrasporreA

Attrs TransposeA(
  bool x
)

TrasporreB

Attrs TransposeB(
  bool x
)