aliran tensor:: operasi:: MatMul

bookmark_border Tetap teratur dengan koleksi Simpan dan kategorikan konten berdasarkan preferensi Anda.

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.

Atribut publik

Operation operation

::tensorflow::Output product

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs TransposeA(
  bool x
)

Attrs TransposeB(
  bool x
)