Trả lời cho sự kiện TensorFlow Everywhere tại địa phương của bạn ngay hôm nay!
Trang này được dịch bởi Cloud Translation API.
Switch to English

dòng chảy :: ops :: MatMul

#include <math_ops.h>

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

Tóm lược

Các đầu vào phải là ma trận hai chiều và kích thước bên trong của "a" (sau khi được chuyển vị nếu transpose_a là true) phải khớp với kích thước bên ngoài của "b" (sau khi được chuyển đổi nếu transposed_b là true).

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

Tranh luận:

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

  • Transpose_a: Nếu đúng, "a" được chuyển vị trước phép nhân.
  • Transpose_b: Nếu đúng, "b" được chuyển vị trước phép nhân.

Lợi nhuận:

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

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

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

TransposeA (bool x)
TransposeB (bool x)

Cấu trúc

tensorflow :: ops :: MatMul :: Attrs

Bộ thiết lập thuộc tính tùy chọn cho MatMul .

Thuộc tính công cộng

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ử :: 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

TransposeA

Attrs TransposeA(
  bool x
)

TransposeB

Attrs TransposeB(
  bool x
)