tensorflow :: 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 ser transpuesta si transpose_a es verdadera) debe coincidir con la dimensión externa de "b" (después de ser transpuesta si transposed_b es verdadera).
Nota : La implementación del kernel predeterminada para MatMul en GPU usa cublas.
Argumentos:
- alcance: un objeto de alcance
Atributos opcionales (consulte 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 publicas | |
---|---|
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 :: ops :: MatMul :: Attrs | Establecedores de atributos opcionales para MatMul . |
Atributos públicos
operación
Operation operation
producto
::tensorflow::Output product
Funciones publicas
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 )
TransponerB
Attrs TransposeB( bool x )