dòng chảy :: ops :: SparseMatMul

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

Tóm lược

Các đầ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 không phải là SparseTensor s. Op này được tối ưu hóa cho trường hợp ít nhất một trong các "a" hoặc "b" là thưa thớt, theo nghĩa là chúng có một tỷ lệ lớn các giá trị 0. Hòa vốn cho việc sử dụng điều này so với một ma trận dày đặc nhân trên một nền tảng là 30% giá trị 0 trong ma trận thưa thớt.

Tính toán gradient của hoạt động này sẽ chỉ tận dụng được sự thưa thớt trong gradient đầu vào khi gradient đó đến từ một Relu .

Tranh luận:

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

Lợi nhuận:

• Output : Bộ căng sản phẩm.

Thuộc tính công cộng

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
)