przepływ tensorowy:: ops:: MatMul
#include <math_ops.h>
Pomnóż macierz „a” przez macierz „b”.
Streszczenie
Dane wejściowe muszą być macierzami dwuwymiarowymi, a wewnętrzny wymiar „a” (po transpozycji, jeśli transpose_a ma wartość true) musi odpowiadać zewnętrznemu wymiarowi „b” (po transpozycji, jeśli transposed_b ma wartość true).
Uwaga : Domyślna implementacja jądra dla MatMul na procesorach graficznych używa cublas.
Argumenty:
- zakres: Obiekt Scope
Opcjonalne atrybuty (patrz Attrs
):
- transpose_a: Jeśli true, „a” jest transponowane przed mnożeniem.
- transpose_b: Jeśli to prawda, „b” jest transponowane przed mnożeniem.
Zwroty:
-
Output
: Tensor iloczynu.
Konstruktory i destruktory | |
---|---|
MatMul (const :: | |
MatMul (const :: |
Funkcje publiczne | |
---|---|
node () const | ::tensorflow::Node * |
operator:: | |
operator:: |
Publiczne funkcje statyczne | |
---|---|
TransposeA (bool x) | |
TransposeB (bool x) |
Struktury | |
---|---|
tensorflow:: ops:: MatMul:: Atrybuty | Opcjonalne ustawiacze atrybutów dla MatMul . |
Atrybuty publiczne
działanie
Operation operation
produkt
::tensorflow::Output product
Funkcje publiczne
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 )