dòng chảy căng:: ôi:: thưa thớtMatMul

Nhân ma trận “a” với ma trận “b”.

Bản tóm tắt

Đầu vào phải là ma trận hai chiều và kích thước bên trong của "a" phải khớp với kích thước bên ngoài của "b". Cả "a" và "b" phải là Tensor chứ không phải SparseTensor . Hoạt động này được tối ưu hóa cho trường hợp ít nhất một trong số "a" hoặc "b" thưa thớt, theo nghĩa là chúng có tỷ lệ lớn các giá trị bằng 0. Mức hòa vốn khi sử dụng giá trị này so với phép nhân ma trận dày đặc trên một nền tảng là 30% giá trị 0 trong ma trận thưa thớt.

Việc tính toán độ dốc của thao tác này sẽ chỉ tận dụng được độ thưa thớt trong độ dốc đầu vào khi độ dốc đó đến từ Relu .

Tranh luận:

• phạm vi: Một đối tượng Phạm vi

Trả về:

• Output : Tenxor tích.

Thuộc tính công khai

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
)