tensorflow::ops::MatMul
#include <math_ops.h>Kalikan matriks "a" dengan matriks "b".
Ringkasan
Input harus berupa matriks dua dimensi dan dimensi dalam dari "a" (setelah ditransposisikan jika transpose_a bernilai true) harus sesuai dengan dimensi luar dari "b" (setelah ditransposisikan jika transposed_b bernilai true).
Catatan : Implementasi kernel default untuk MatMul pada GPU menggunakan cublas.
Argumen:
- Cakupan: Objek Cakupan
Atribut opsional (lihat Attrs ):
- transpose_a: Jika benar, "a" ditransposisikan sebelum perkalian.
- transpose_b: Jika bernilai true, "b" akan ditransposisikan sebelum perkalian.
Mengembalikan:
-
Output: Tensor produk.
Konstruktor dan Destruktor | |
|---|---|
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) |
Atribut publik | |
|---|---|
operation | |
product | |
Fungsi publik | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
Fungsi statis publik | |
|---|---|
TransposeA (bool x) | |
TransposeB (bool x) | |
Struktur | |
|---|---|
| tensorflow::ops::MatMul::Attrs | Pengaturan atribut opsional untuk MatMul . |
Atribut publik
operasi
Operation operation
produk
::tensorflow::Output product
Fungsi publik
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 )
simpul
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
Fungsi statis publik
TransposeA
Attrs TransposeA( bool x )
TransposeB
Attrs TransposeB( bool x )