tensorflow::ops::BatchMatMul

Nhân các lát cắt của hai tensor theo từng lô.

Bản tóm tắt

Hàm này nhân tất cả các lát cắt của Tensor x và y (mỗi lát cắt có thể được xem như một phần tử của một batch), và sắp xếp các kết quả riêng lẻ thành một tensor đầu ra duy nhất có cùng kích thước batch. Mỗi lát cắt riêng lẻ có thể được phối hợp (phối hợp ma trận có nghĩa là chuyể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 , mặc định là False .

Các tensor đầu vào x và y là tensor 2 chiều trở lên với hình dạng [..., r_x, c_x] và [..., r_y, c_y] .

Tensor đầu ra là tensor 2 chiều trở lê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[..., :, :])

Luận điểm:

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

Các thuộc tính tùy chọn (xem Attrs ):

• adj_x: Nếu True , sẽ liên kết các lát cắt của x . Mặc định là False .
• adj_y: Nếu True , sẽ liên kết các lát cắt của y . Mặc định là False .

Trả về:

• Output : Hình ảnh 3D 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
)