dòng chảy căng:: ôi:: MatMul

#include <math_ops.h>

Nhân ma trận "a" với ma trận "b".

Bản tóm tắt

Dữ liệu đầu vào phải là ma trận hai chiều và chiều bên trong của "a" (sau khi được chuyển đổi nếu transpose_a là đúng) phải khớp với chiều bên ngoài của "b" (sau khi được chuyển đổi nếu transpose_b là đúng).

Lưu ý : Việc triển khai kernel mặc định cho MatMul trên GPU sử dụng cublas.

Lập luận:

• phạm vi: Một đối tượng Phạm vi

Thuộc tính tùy chọn (xem Attrs ):

• transpose_a: Nếu đúng, "a" được chuyển vị trước khi nhân.
• transpose_b: Nếu đúng, "b" được chuyển vị trước khi nhân.

Trả về:

• Output : Tenxor tích.

Hàm tạo và hàm hủy
MatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b)
MatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b, const MatMul::Attrs & attrs)

Thuộc tính công khai
operation	Operation
product	:: tensorflow::Output

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
TransposeA (bool x)	Attrs
TransposeB (bool x)	Attrs

Cấu trúc
tenorflow:: ops:: MatMul:: Attrs	Bộ cài đặt thuộc tính tùy chọn cho MatMul .

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

MatMul

 MatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b
)

MatMul

 MatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a,
  ::tensorflow::Input b,
  const MatMul::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

Chuyển vịA

Attrs TransposeA(
  bool x
)

Chuyển đổiB

Attrs TransposeB(
  bool x
)