텐서플로우:: 작전:: 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"가 전치됩니다.
보고:
-
Output
: 제품 텐서.
생성자와 소멸자 | |
---|---|
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 )