ML 커뮤니티 데이는 11월 9일입니다! TensorFlow, JAX에서 업데이트를 우리와 함께, 더 자세히 알아보기

tensorflow :: ops :: BatchMatMul

#include <math_ops.h>

두 개의 텐서 조각을 일괄 적으로 곱합니다.

요약

Tensor xy 의 모든 슬라이스를 곱하고 (각 슬라이스는 배치의 요소로 볼 수 있음) 개별 결과를 동일한 배치 크기의 단일 출력 텐서로 정렬합니다. 각 슬라이스마다 임의로 adjointed 수 곱하기 전에 설정 (전치과 공역 할 수있는 매트릭스 수단 수반 행렬 위해) adj_x 또는 adj_y 에 플래그 True 기본으로하고, False .

입력 텐서 xy 는 모양이 [..., 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[..., :, :])

인수 :

  • 범위 : 범위 개체
  • x : 2 차원 이상, 모양 [..., r_x, c_x] .
  • y : 모양 [..., r_y, c_y] 2 차원 이상.

선택적 속성 ( Attrs 참조) :

  • adj_x : True 이면 x 조각을 인접합니다. 기본값은 False 입니다.
  • adj_y : True 이면 y 의 조각을 연결합니다. 기본값은 False 입니다.

보고:

  • Output : 3 차원 이상, 모양 [..., r_o, c_o]

생성자와 소멸자

BatchMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y)
BatchMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y, const BatchMatMul::Attrs & attrs)

공용 속성

operation
output

공공 기능

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

공개 정적 함수

AdjX (bool x)
AdjY (bool x)

구조체

tensorflow :: ops :: BatchMatMul :: Attrs

BatchMatMul에 대한 선택적 속성 설정자.

공용 속성

조작

Operation operation

산출

::tensorflow::Output output

공공 기능

BatchMatMul

 BatchMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y
)

BatchMatMul

 BatchMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y,
  const BatchMatMul::Attrs & attrs
)

마디

::tensorflow::Node * node() const 

연산자 :: tensorflow :: 입력

 operator::tensorflow::Input() const 

연산자 :: tensorflow :: 출력

 operator::tensorflow::Output() const 

공개 정적 함수

AdjX

Attrs AdjX(
  bool x
)

AdjY

Attrs AdjY(
  bool x
)