tensorflow :: ops :: SparseMatMul

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

요약

입력은 2 차원 행렬이어야하며 "a"의 내부 차원은 "b"의 외부 차원과 일치해야합니다. "a"와 "b"는 모두 SparseTensor 아닌 Tensor 여야합니다. 이 연산은 "a"또는 "b"중 적어도 하나가 0 값의 비율이 크다는 점에서 희소 한 경우에 최적화됩니다. 한 플랫폼에서 이것을 사용하는 손익분기 점 대 조밀 행렬 곱셈은 희소 행렬에서 30 % 0 값이었습니다.

이 연산의 그래디언트 계산은 그래디언트가 Relu 에서 나올 때 입력 그래디언트의 희소성을 이용합니다.

인수 :

• 범위 : 범위 개체

보고:

• Output : 제품 텐서.

공용 속성

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 

Attrs AIsSparse(
  bool x
)

Attrs BIsSparse(
  bool x
)

Attrs TransposeA(
  bool x
)

Attrs TransposeB(
  bool x
)