tensorflow :: ops :: MatMul

Pomnóż macierz „a” przez macierz „b”.

streszczenie

Dane wejściowe muszą być dwuwymiarowymi macierzami, a wewnętrzny wymiar „a” (po transpozycji, jeśli transpose_a ma wartość prawda) musi odpowiadać zewnętrznemu wymiarowi „b” (po transpozycji, jeśli transposed_b ma wartość prawda).

Uwaga : domyślna implementacja jądra dla MatMul na GPU używa cublas.

Argumenty:

• zakres: obiekt Scope

Atrybuty opcjonalne (patrz Attrs ):

• transpose_a: Jeśli prawda, „a” jest transponowane przed pomnożeniem.
• transpose_b: Jeśli prawda, „b” jest transponowane przed pomnożeniem.

Zwroty:

• Output : tensor produktu.

Atrybuty publiczne

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
)