텐서플로우:: 작전:: 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:: 속성

MatMul 에 대한 선택적 속성 설정자.

공개 속성

작업

Operation operation

제품

::tensorflow::Output product

공공 기능

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
)

마디

::tensorflow::Node * node() const 

연산자::텐서플로우::입력

 operator::tensorflow::Input() const 

연산자::텐서플로우::출력

 operator::tensorflow::Output() const 

공개 정적 함수

조옮김A

Attrs TransposeA(
  bool x
)

조옮김B

Attrs TransposeB(
  bool x
)