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:

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:

Konstruktory i destruktory

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)

Atrybuty publiczne

operation
product

Funkcje publiczne

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Publiczne funkcje statyczne

TransposeA (bool x)
TransposeB (bool x)

Struktury

tensorflow:: ops:: MatMul:: Atrybuty

Opcjonalne ustawiacze atrybutów dla MatMul .

Atrybuty publiczne

operacja

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
)

węzeł

::tensorflow::Node * node() const

operator::tensorflow::Wejście

 operator::tensorflow::Input() const

operator::tensorflow::Wyjście

 operator::tensorflow::Output() const

Publiczne funkcje statyczne

TranspozycjaA

Attrs TransposeA(
  bool x
)

Transpozycja B

Attrs TransposeB(
  bool x
)