공개 최종 수업 MatMul
행렬 "a"에 행렬 "b"를 곱합니다.
입력은 2차원 행렬이어야 하며 "a"의 내부 차원(transpose_a가 true인 경우 전치된 후)은 "b"의 외부 차원(transposed_b가 true인 경우 전치된 후)과 일치해야 합니다.
참고 : GPU에서 MatMul의 기본 커널 구현은 cublas를 사용합니다.
중첩 클래스
| 수업 | MatMul.옵션 | MatMul 의 선택적 속성 | |
상수
| 끈 | OP_NAME | TensorFlow 코어 엔진에서 알려진 이 작업의 이름 |
공개 방법
| 출력 <T> | 출력 () 텐서의 기호 핸들을 반환합니다. |
| static <T는 TType을 확장합니다. > MatMul <T> | |
| 출력 <T> | 제품 () |
| 정적 MatMul.Options | transposeA (부울 전치 A) |
| 정적 MatMul.Options | transposeB (부울 전치 B) |
상속된 메서드
상수
공개 정적 최종 문자열 OP_NAME
TensorFlow 코어 엔진에서 알려진 이 작업의 이름
상수 값: "MatMul"
공개 방법
공개 출력 <T> asOutput ()
텐서의 기호 핸들을 반환합니다.
TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.
public static MatMul <T> create ( 범위 범위, 피연산자 <T> a, 피연산자 <T> b, 옵션... 옵션)
새로운 MatMul 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
| 범위 | 현재 범위 |
|---|---|
| 옵션 | 선택적 속성 값을 전달합니다. |
보고
- MatMul의 새로운 인스턴스