MatMul

공개 최종 수업 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> a, 피연산자 <T> b, 옵션... 옵션)
새로운 MatMul 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <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의 새로운 인스턴스

공개 출력 <T> 제품 ()

공개 정적 MatMul.Options transposeA (부울 transposeA)

매개변수
전치 A true인 경우 "a"는 곱하기 전에 전치됩니다.

공개 정적 MatMul.Options transposeB (부울 전치 B)

매개변수
B true인 경우 "b"는 곱하기 전에 전치됩니다.