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ận SparseTensor , kích thước [nnz, 2] .
  • a_giá trị: 1-D. Các values của SparseTensor , kích thước [nnz] Vector.
  • a_shape: 1-D. shape của SparseTensor , 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ề:

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
)