tensorflow::ops::SparseMatMul

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à chiều trong của "a" phải khớp với chiều ngoài của "b". Cả "a" và "b" đều phải là Tensor chứ không phải SparseTensor . Phép toán này được tối ưu hóa cho trường hợp ít nhất một trong "a" hoặc "b" là ma trận thưa, theo nghĩa là chúng có tỷ lệ giá trị bằng 0 cao. Điểm hòa vốn khi sử dụng phép toán 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ị bằng 0 trong ma trận thưa.

Việc tính toán đạo hàm của phép toán này chỉ tận dụng được tính thưa thớt trong đạo hàm đầu vào khi đạo hàm đó đến từ hàm ReLU .

Luận điểm:

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

Trả về:

• Output : Tensor 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
)