dòng chảy căng:: ôi:: hàng loạtMatMul

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 x và y (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 nối 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 x và y là 2-D hoặc cao hơn với hình dạng [..., r_x, c_x] và [..., r_y, c_y] .

Tenxor đầ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 là:

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

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

Thuộc tính công khai

Operation operation

::tensorflow::Output output

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

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs AdjX(
  bool x
)

Attrs AdjY(
  bool x
)