MatMul

lớp cuối cùng công khai MatMul

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

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.

Các lớp lồng nhau

lớp học MatMul.Options Thuộc tính tùy chọn cho MatMul

Hằng số

Sợi dây OP_NAME Tên của op này, được biết đến bởi công cụ lõi TensorFlow

Phương pháp công cộng

Đầu ra <T>
asOutput ()
Trả về tay cầm tượng trưng của tensor.
tĩnh <T mở rộng TType > MatMul <T>
tạo ( Phạm vi phạm vi, Toán hạng <T> a, Toán hạng <T> b, Tùy chọn... tùy chọn)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác MatMul mới.
Đầu ra <T>
MatMul.Options tĩnh
chuyển vịA (chuyển đổi BooleanA)
MatMul.Options tĩnh
transposeB (chuyển đổi BooleanB)

Phương pháp kế thừa

Hằng số

Chuỗi cuối cùng tĩnh công khai OP_NAME

Tên của op này, được biết đến bởi công cụ lõi TensorFlow

Giá trị không đổi: "MatMul"

Phương pháp công cộng

Đầu ra công khai <T> asOutput ()

Trả về tay cầm tượng trưng của tensor.

Đầu vào của các hoạt động TensorFlow là đầu ra của một hoạt động TensorFlow khác. Phương pháp này được sử dụng để thu được một thẻ điều khiển mang tính biểu tượng đại diện cho việc tính toán đầu vào.

public static MatMul <T> create ( Phạm vi phạm vi, Toán hạng <T> a, Toán hạng <T> b, Tùy chọn... tùy chọn)

Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác MatMul mới.

Thông số
phạm vi phạm vi hiện tại
tùy chọn mang các giá trị thuộc tính tùy chọn
Trả lại
  • một phiên bản mới của MatMul

Sản phẩm công khai <T> sản phẩm ()

public static MatMul.Options transposeA (Boolean transposeA)

Thông số
chuyển vịA Nếu đúng, "a" được hoán vị trước khi nhân.

công khai MatMul.Options transposeB (Boolean transposeB)

Thông số
chuyển vịB Nếu đúng, "b" được hoán vị trước khi nhân.