Nhân ma trận “a” với ma trận “b”.
Đầ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.
Các lớp lồng nhau
lớp học | SparseMatMul.Options | Thuộc tính tùy chọn cho SparseMatMul |
Hằng số
Sợi dây | OP_NAME | Tên của op này, được biết đến bởi công cụ lõi TensorFlow |
Phương pháp công khai
tĩnh SparseMatMul.Options | aIsSparse (Boolean aIsSparse) |
Đầu ra < TFloat32 > | asOutput () Trả về tay cầm tượng trưng của tensor. |
tĩnh SparseMatMul.Options | bIsSparse (Boolean bIsSparse) |
tĩnh thưa thớtMul | |
Đầu ra < TFloat32 > | sản phẩm () |
tĩnh SparseMatMul.Options | chuyển vịA (chuyển đổi BooleanA) |
tĩnh SparseMatMul.Options | transposeB (chuyển đổi BooleanB) |
Phương pháp kế thừa
Hằng số
Chuỗi cuối cùng tĩnh công khai OP_NAME
Tên của op này, được biết đến bởi công cụ lõi TensorFlow
Phương pháp công khai
Đầu ra công khai < TFloat32 > asOutput ()
Trả về tay cầm tượng trưng của tensor.
Đầu vào của các hoạt động TensorFlow là đầu ra của một hoạt động TensorFlow khác. Phương pháp này được sử dụng để thu được một thẻ điều khiển mang tính biểu tượng đại diện cho việc tính toán đầu vào.
public static SparseMatMul tạo ( Phạm vi phạm vi, Toán hạng <? mở rộng TNumber > a, Toán hạng <? mở rộng TNumber > b, Tùy chọn... tùy chọn)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác SparseMatMul mới.
Thông số
phạm vi | phạm vi hiện tại |
---|---|
tùy chọn | mang các giá trị thuộc tính tùy chọn |
Trả lại
- một phiên bản mới của SparseMatMul