텐서플로우:: 작전:: 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

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

SparseMatMul

 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
)

BIs스파스

Attrs BIsSparse(
  bool x
)

조옮김A

Attrs TransposeA(
  bool x
)

조옮김B

Attrs TransposeB(
  bool x
)