dòng chảy :: hoạt động :: 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 nhân 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

Các 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

Các chức năng tĩnh công khai

TransposeA

Attrs TransposeA(
  bool x
)

TransposeB

Attrs TransposeB(
  bool x
)