dòng chảy :: hoạt động :: SparseTensorDenseMatMul

#include <sparse_ops.h>

Nhân SparseTensor (thuộc hạng 2) "A" với ma trận dày đặc "B".

Tóm lược

Không có kiểm tra tính hợp lệ nào được thực hiện trên các chỉ số của A. Tuy nhiên, định dạng đầu vào sau được khuyến nghị cho hành vi tối ưu:

if adjoint_a == false: A nên được sắp xếp theo thứ tự tăng dần về mặt từ vựng. 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 thứ nguyên 1 (tức là thứ tự "cột chính" thay vì thứ tự "hàng chính").

Tranh luận:

  • phạm vi: Một đối tượng Phạm vi
  • a_indices: 2-D. Các indices của Ma trận SparseTensor , size [nnz, 2] .
  • a_values: 1-D. Các values của SparseTensor , size [nnz] .
  • a_shape: 1-D. Hình shape của SparseTensor , kích thước [2] Vector.
  • b: 2-D. Ma trận dày đặc.

Các thuộc tính tùy chọn (xem Phần Attrs ):

  • adjoint_a: Sử dụng adjoint_a trong phép nhân ma trận. Nếu A là phức, đây là chuyển vị (obs (A)). Nếu không thì nó chuyển vị (A).
  • adjoint_b: Sử dụng adjoint của B trong phép nhân ma trận. Nếu B là phức, đây là chuyển vị (obs (B)). Nếu không thì nó chuyển vị (B).

Lợi nhuận:

  • Output : Bộ căng sản phẩm.

Người xây dựng và Người phá 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

Chức năng tĩnh công khai

AdjointA (bool x)
AdjointB (bool x)

Cấu trúc

tensorflow :: ops :: SparseTensorDenseMatMul :: Attrs

Bộ thiết lập 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

SparseTensorDenseMatMul

 SparseTensorDenseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a_indices,
  ::tensorflow::Input a_values,
  ::tensorflow::Input a_shape,
  ::tensorflow::Input b
)

SparseTensorDenseMatMul

 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ử :: tensorflow :: Đầu vào

 operator::tensorflow::Input() const

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

 operator::tensorflow::Output() const

Chức năng tĩnh công khai

AdjointA

Attrs AdjointA(
  bool x
)

AdjointB

Attrs AdjointB(
  bool x
)