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:

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

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

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
)