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

#include <math_ops.h>

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 .

Lập luận:

Trả về:

Hàm tạo và hàm hủy

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)

Thuộc tính công khai

operation
product

Chức năng công cộng

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Các hàm tĩnh công khai

AIsSparse (bool x)
BIsSparse (bool x)
TransposeA (bool x)
TransposeB (bool x)

Cấu trúc

tenorflow:: ops:: SparseMatMul:: Attrs

Bộ cài đặt thuộc tính tùy chọn cho SparseMatMul .

Thuộc tính công khai

hoạt động 

Operation operation

sản phẩm 

::tensorflow::Output product

Chức năng công cộng

thưa thớtMatMul 

 SparseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b
)

thưa thớtMatMul 

 SparseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b,
  const SparseMatMul::Attrs & attrs
)

nút 

::tensorflow::Node * node() const

toán tử::tenorflow::Đầu vào 

 operator::tensorflow::Input() const

toán tử::tenorflow::Đầu ra 

 operator::tensorflow::Output() const

Các hàm tĩnh công khai

AIsthưa thớt 

Attrs AIsSparse(
  bool x
)

BIsthưa thớt 

Attrs BIsSparse(
  bool x
)

Chuyển vịA 

Attrs TransposeA(
  bool x
)

Chuyển đổiB 

Attrs TransposeB(
  bool x
)