텐서플로우:: ops:: SparseMatMul

#include <math_ops.h>

행렬 "a"와 행렬 "b"를 곱합니다 .

요약

입력은 2차원 행렬이어야 하며, "a"의 내부 차원은 "b"의 외부 차원과 일치해야 합니다. "a"와 "b"는 모두 SparseTensor 가 아닌 Tensor 여야 합니다. 이 연산은 "a" 또는 "b" 중 하나 이상이 희소 행렬(즉, 0의 비율이 높은 행렬)인 경우에 최적화되어 있습니다. 한 플랫폼에서 이 연산을 사용하는 것과 밀집 행렬 곱셈을 사용하는 것의 손익분기점은 희소 행렬에서 0의 비율이 30%일 때였습니다.

이 연산의 기울기 계산은 입력 기울기가 ReLU 에서 나온 경우에만 입력 기울기의 희소성을 활용합니다.

논거:

보고:

생성자와 소멸자

SparseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b)
SparseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b, const SparseMatMul::Attrs & attrs)

공개 속성

operation
product

공공 기능

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

공개 정적 함수

AIsSparse (bool x)
BIsSparse (bool x)
TransposeA (bool x)
TransposeB (bool x)

구조

텐서플로우:: ops:: SparseMatMul:: Attrs

SparseMatMul 에 대한 선택적 속성 설정기입니다.

공개 속성

작업 

Operation operation

제품 

::tensorflow::Output product

공공 기능

희소매트멀 

 SparseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b
)

희소매트멀 

 SparseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b,
  const SparseMatMul::Attrs & attrs
)

마디 

::tensorflow::Node * node() const

연산자::텐서플로우::입력 

 operator::tensorflow::Input() const

연산자::텐서플로우::출력 

 operator::tensorflow::Output() const

공개 정적 함수

AIsSparse 

Attrs AIsSparse(
  bool x
)

BIsSparse 

Attrs BIsSparse(
  bool x
)

전치A 

Attrs TransposeA(
  bool x
)

전치B 

Attrs TransposeB(
  bool x
)