Google I/O에 참여해 주셔서 감사합니다. 주문형 모든 세션 보기 주문형 시청

tensorflow :: ops :: MatMul

#include <math_ops.h>

행렬 "a"에 행렬 "b"를 합니다.

요약

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

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

인수 :

선택적 속성 ( Attrs 참조) :

  • transpose_a : 참이면 곱하기 전에 "a"가 전치됩니다.
  • transpose_b : 참이면 곱하기 전에 "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)

구조체

tensorflow :: 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 

연산자 :: tensorflow :: 입력

 operator::tensorflow::Input() const 

연산자 :: tensorflow :: 출력

 operator::tensorflow::Output() const 

공개 정적 함수

TransposeA

Attrs TransposeA(
  bool x
)

조옮김 B

Attrs TransposeB(
  bool x
)