flujo tensor:: operaciones:: matmul

#include <math_ops.h>

Multiplica la matriz "a" por la matriz "b".

Resumen

Las entradas deben ser matrices bidimensionales y la dimensión interna de "a" (después de transponerse si transpose_a es verdadera) debe coincidir con la dimensión externa de "b" (después de transponerse si transpose_b es verdadera).

Nota : La implementación predeterminada del kernel para MatMul en GPU utiliza cubelas.

Argumentos:

Atributos opcionales (ver Attrs ):

  • transpose_a: si es verdadero, "a" se transpone antes de la multiplicación.
  • transpose_b: si es verdadero, "b" se transpone antes de la multiplicación.

Devoluciones:

  • Output : El tensor del producto.

Constructores y destructores

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)

Atributos públicos

operation
product

Funciones públicas

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

Funciones estáticas públicas

TransposeA (bool x)
TransposeB (bool x)

estructuras

tensorflow:: operaciones:: MatMul:: Atributos

Configuradores de atributos opcionales para MatMul .

Atributos públicos

operación

Operation operation

producto

::tensorflow::Output product

Funciones públicas

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 

operador::tensorflow::Entrada

 operator::tensorflow::Input() const 

operador::tensorflow::Salida

 operator::tensorflow::Output() const 

Funciones estáticas públicas

TransponerA

Attrs TransposeA(
  bool x
)

Transponer B

Attrs TransposeB(
  bool x
)