tensorflow :: ops :: SparseMatMul

"A" matrisini "b" matrisiyle çarpın .

Özet

Girdiler iki boyutlu matrisler olmalı ve "a" nın iç boyutu "b" nin dış boyutuyla eşleşmelidir. Hem "a" hem de "b" Tensor olmalı, SparseTensor değil. Bu işlem, "a" veya "b" den en az birinin, büyük oranda sıfır değerine sahip olmaları bakımından seyrek olduğu durum için optimize edilmiştir. Bunu bir platformda yoğun matris çarpımına karşı kullanmak için başabaş, seyrek matriste% 30 sıfır değerleriydi.

Bu işlemin gradyan hesaplaması, yalnızca gradyan bir Relu'dan geldiğinde giriş gradyanındaki seyreklikten yararlanacaktır.

Argümanlar:

• kapsam: Bir Scope nesnesi

İadeler:

• Output : Ürün tensörü.

Genel özellikler

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
)