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
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 cắ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]
.
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ủax
. Mặc định làFalse
. - adj_y: Nếu
True
, hãy nối các lát cắt củay
. 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 )