dòng chảy :: hoạt động :: BatchMatMul

#include <math_ops.h>

Nhân các lát của hai tenxơ theo lô.

Tóm lược

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 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 phòng trong số lát cá nhân có thể tùy chọn được adjointed (để liên hợp là một phương tiện ma trận để transpose và liên hợp nó) trước khi nhân bằng cách thiết lập adj_x hoặc adj_y cờ True , đó là theo mặc định False .

Các hàng chục đầ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] .

Bộ căng đầ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 hoặc cao hơn với hình dạng [..., r_x, c_x] .
  • y: 2-D hoặc cao hơn với hình dạng [..., r_y, c_y] .

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

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

Lợi nhuận:

  • Output : 3-D hoặc cao hơn với hình dạng [..., r_o, c_o]

Người xây dựng và Người phá 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 cộng

operation
output

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

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Các chức năng tĩnh công khai

AdjX (bool x)
AdjY (bool x)

Cấu trúc

tensorflow :: ops :: BatchMatMul :: Attrs

Bộ thiết lập thuộc tính tùy chọn cho BatchMatMul .

Thuộc tính công cộng

hoạt động

Operation operation

đầu ra

::tensorflow::Output output

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

BatchMatMul

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

BatchMatMul

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

nút

::tensorflow::Node * node() const

toán tử :: tensorflow :: Đầu vào

 operator::tensorflow::Input() const

toán tử :: tensorflow :: Đầu ra

 operator::tensorflow::Output() const

Các chức năng tĩnh công khai

AdjX

Attrs AdjX(
  bool x
)

Điều chỉnhY

Attrs AdjY(
  bool x
)