dòng chảy căng:: ôi:: Thưa thớtTensorDenseMatMul
#include <sparse_ops.h>
Nhân SparseTensor (hạng 2) "A" với ma trận dày đặc "B".
Bản tóm tắt
Không có kiểm tra tính hợp lệ nào được thực hiện trên các chỉ mục của A. Tuy nhiên, định dạng đầu vào sau được khuyến nghị để có hành vi tối ưu:
if adjoint_a == false: A phải được sắp xếp theo thứ tự từ điển tăng dần. Sử dụng SparseReorder nếu bạn không chắc chắn. if adjoint_a == true: A phải được sắp xếp theo thứ tự tăng dần 1 (tức là thứ tự "cột chính" thay vì thứ tự "hàng chính").
Lập luận:
- phạm vi: Một đối tượng Phạm vi
- a_chỉ số: 2-D.
indices
của Ma trậnSparseTensor
, kích thước[nnz, 2]
. - a_giá trị: 1-D. Các
values
củaSparseTensor
, kích thước[nnz]
Vector. - a_shape: 1-D.
shape
củaSparseTensor
, kích thước[2]
Vector. - b: 2-D. Một ma trận dày đặc.
Thuộc tính tùy chọn (xem Attrs
):
- adjoint_a: Sử dụng liền kề của A trong ma trận nhân. Nếu A phức thì đây là transpose(conj(A)). Nếu không thì đó là chuyển vị (A).
- adjoint_b: Sử dụng liền kề của B trong ma trận nhân. Nếu B phức tạp thì đây là transpose(conj(B)). Nếu không thì đó là chuyển vị (B).
Trả về:
-
Output
: Tenxor tích.
Hàm tạo và hàm hủy | |
---|---|
SparseTensorDenseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a_indices, :: tensorflow::Input a_values, :: tensorflow::Input a_shape, :: tensorflow::Input b) | |
SparseTensorDenseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a_indices, :: tensorflow::Input a_values, :: tensorflow::Input a_shape, :: tensorflow::Input b, const SparseTensorDenseMatMul::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 | |
---|---|
AdjointA (bool x) | |
AdjointB (bool x) |
Cấu trúc | |
---|---|
tensorflow:: ops:: SparseTensorDenseMatMul:: Attrs | Bộ cài đặt thuộc tính tùy chọn cho SparseTensorDenseMatMul . |
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ớtTensorDenseMatMul
SparseTensorDenseMatMul( const ::tensorflow::Scope & scope, ::tensorflow::Input a_indices, ::tensorflow::Input a_values, ::tensorflow::Input a_shape, ::tensorflow::Input b )
Thưa thớtTensorDenseMatMul
SparseTensorDenseMatMul( const ::tensorflow::Scope & scope, ::tensorflow::Input a_indices, ::tensorflow::Input a_values, ::tensorflow::Input a_shape, ::tensorflow::Input b, const SparseTensorDenseMatMul::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
Liên kết A
Attrs AdjointA( bool x )
Liên kếtB
Attrs AdjointB( bool x )