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
)