tensorflow::ops::MatMul

#include <math_ops.h>

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

Bản tóm tắt

Các đầu vào phải là ma trận hai chiều và chiều trong của "a" (sau khi được chuyển vị nếu transpose_a là true) phải khớp với chiều ngoài của "b" (sau khi được chuyển vị 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.

Luận điểm:

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

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

Trả về:

Hàm khởi 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
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

TransposeA (bool x)
TransposeB (bool x)

Cấu trúc

tensorflow::ops::MatMul::Attrs

Các thiết lập 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

operator::tensorflow::Input 

 operator::tensorflow::Input() const

operator::tensorflow::Output 

 operator::tensorflow::Output() const

Các hàm tĩnh công khai

Chuyển vị A 

Attrs TransposeA(
  bool x
)

Chuyển vị B 

Attrs TransposeB(
  bool x
)