aliran tensor:: operasi:: MatMul
#include <math_ops.h>
Kalikan matriks "a" dengan matriks "b".
Ringkasan
Inputnya harus berupa matriks dua dimensi dan dimensi dalam "a" (setelah ditransposisikan jika transpose_a benar) harus sesuai dengan dimensi luar "b" (setelah ditransposisikan jika transpose_b benar).
Catatan : Implementasi kernel default untuk MatMul pada GPU menggunakan culas.
Argumen:
- ruang lingkup: Objek Lingkup
Atribut opsional (lihat Attrs
):
- transpose_a: Jika benar, "a" dialihkan sebelum perkalian.
- transpose_b: Jika benar, "b" dialihkan sebelum perkalian.
Pengembalian:
-
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 | Penyetel 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::Masukan
operator::tensorflow::Input() const
operator::tensorflow::Keluaran
operator::tensorflow::Output() const
Fungsi statis publik
Mengubah urutanA
Attrs TransposeA( bool x )
TransposB
Attrs TransposeB( bool x )