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
)