텐서플로우:: 작전:: BatchMatMulV2
#include <math_ops.h>
두 개의 텐서 조각을 일괄적으로 곱합니다.
요약
Tensor
x
와 y
의 모든 조각을 곱하고(각 조각은 배치의 요소로 볼 수 있음) 동일한 배치 크기의 단일 출력 텐서에 개별 결과를 정렬합니다. adj_x
또는 adj_y
플래그를 True
로 설정하여(기본적으로 False
) 곱하기 전에 각 개별 조각을 선택적으로 인접시킬 수 있습니다(행렬을 인접한다는 것은 행렬을 전치하고 공액시키는 것을 의미합니다).
입력 텐서 x
및 y
는 [..., r_x, c_x]
및 [..., r_y, c_y]
모양의 2차원 이상입니다.
출력 텐서는 [..., r_o, c_o]
모양의 2차원 이상입니다. 여기서:
r_o = c_x if adj_x else r_x c_o = r_y if adj_y else c_y
다음과 같이 계산됩니다.
output[..., :, :] = matrix(x[..., :, :]) * matrix(y[..., :, :])
참고 : BatchMatMulV2
배치 차원의 브로드캐스팅을 지원합니다. 방송에 대한 자세한 내용은 여기를 참조하세요.
인수:
- 범위: 범위 개체
- x:
[..., r_x, c_x]
모양의 2차원 이상. - y:
[..., r_y, c_y]
모양의 2차원 이상.
선택적 속성( Attrs
참조):
- adj_x:
True
인 경우x
조각을 인접시킵니다. 기본값은False
입니다. - adj_y:
True
이면y
의 조각을 인접시킵니다. 기본값은False
입니다.
보고:
-
Output
:[..., r_o, c_o]
모양의 3차원 이상
생성자와 소멸자 | |
---|---|
BatchMatMulV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y) | |
BatchMatMulV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y, const BatchMatMulV2::Attrs & attrs) |
공공 기능 | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
구조체 | |
---|---|
텐서플로우:: ops:: BatchMatMulV2:: 속성 | BatchMatMulV2 에 대한 선택적 속성 설정자입니다. |
공개 속성
작업
Operation operation
산출
::tensorflow::Output output
공공 기능
BatchMatMulV2
BatchMatMulV2( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input y )
BatchMatMulV2
BatchMatMulV2( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input y, const BatchMatMulV2::Attrs & attrs )