텐서플로우:: ops:: MatMul

#include <math_ops.h>

행렬 "a"와 행렬 "b"를 곱합니다 .

요약

입력값은 2차원 행렬이어야 하며, "a" 행렬의 내부 차원(transpose_a가 true인 경우 전치 후)은 "b" 행렬의 외부 차원(transposed_b가 true인 경우 전치 후)과 일치해야 합니다.

참고 : GPU에서 MatMul을 위한 기본 커널 구현은 cublas를 사용합니다.

논거:

선택적 속성( Attrs 참조):

  • transpose_a: true로 설정하면 곱셈 전에 "a"를 전치합니다.
  • transpose_b: true로 설정하면 곱셈 전에 "b"를 전치합니다.

보고:

생성자와 소멸자

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)

공개 속성

operation
product

공공 기능

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

공개 정적 함수

TransposeA (bool x)
TransposeB (bool x)

구조

텐서플로우:: ops:: MatMul:: Attrs

MatMul 에 대한 선택적 속성 설정기입니다.

공개 속성

작업

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 

공개 정적 함수

전치A

Attrs TransposeA(
  bool x
)

전치B

Attrs TransposeB(
  bool x
)