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

#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ậ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

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)

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:: BatchMatMul:: Attrs

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

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

hàng loạtMatMul

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

hàng loạtMatMul

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