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:

Atribut opsional (lihat Attrs ):

  • transpose_a: Jika benar, "a" dialihkan sebelum perkalian.
  • transpose_b: Jika benar, "b" dialihkan sebelum perkalian.

Pengembalian:

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
)