dòng chảy căng:: ôi:: LôMatMulV2

#include <math_ops.h>

Nhân các lát cắt của hai tensor theo đợt.

Bản tóm tắt

Nhân tất cả các lát của Tensor xy (mỗi lát có thể được xem như một phần tử của một lô) và sắp xếp các kết quả riêng lẻ trong một tensor đầu ra duy nhất có cùng kích thước lô. Mỗi lát riêng lẻ có thể được ghép tùy ý (nối một ma trận có nghĩa là hoán vị và liên hợp nó) trước khi nhân bằng cách đặt cờ adj_x hoặc adj_y thành True , theo mặc định là False .

Các tensor đầu vào xy là 2-D hoặc cao hơn với hình dạng [..., r_x, c_x][..., r_y, c_y] .

Tenor đầu ra là 2-D hoặc cao hơn với hình dạng [..., r_o, c_o] , trong đó:

r_o = c_x if adj_x else r_x
c_o = r_y if adj_y else c_y

Nó được tính như sau:

output[..., :, :] = matrix(x[..., :, :]) * matrix(y[..., :, :])

LƯU Ý : BatchMatMulV2 hỗ trợ phát sóng theo kích thước lô. Thông tin thêm về phát sóng ở đây .

Lập luận:

  • phạm vi: Một đối tượng Phạm vi
  • x: 2-D trở lên với hình dạng [..., r_x, c_x] .
  • y: 2-D trở lên với hình dạng [..., r_y, c_y] .

Thuộc tính tùy chọn (xem Attrs ):

  • adj_x: Nếu True , hãy nối các lát của x . Mặc định là False .
  • adj_y: Nếu True , hãy nối các lát cắt của y . Mặc định là False .

Trả về:

  • Output : 3-D trở lên với hình dạng [..., r_o, c_o]

Hàm tạo và hàm hủy

BatchMatMulV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y)
BatchMatMulV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input y, const BatchMatMulV2::Attrs & attrs)

Thuộc tính công khai

operation
output

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

AdjX (bool x)
AdjY (bool x)

Cấu trúc

tenorflow:: ops:: BatchMatMulV2:: Attrs

Bộ cài đặt thuộc tính tùy chọn cho BatchMatMulV2 .

Thuộc tính công khai

hoạt động

Operation operation

đầu ra

::tensorflow::Output output

Chức năng công cộng

LôMatMulV2

 BatchMatMulV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y
)

LôMatMulV2

 BatchMatMulV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y,
  const BatchMatMulV2::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

AdjX

Attrs AdjX(
  bool x
)

điều chỉnh

Attrs AdjY(
  bool x
)