tensorflow::ops::Diag
#include <array_ops.h>Trả về một tensor đường chéo với các giá trị đường chéo đã cho.
Bản tóm tắt
Cho một diagonal , phép toán này trả về một tensor với diagonal và tất cả các phần còn lại được đệm bằng số 0. Đường chéo được tính như sau:
Giả sử diagonal có kích thước [D1,..., Dk], thì đầu ra là một tensor hạng 2k có kích thước [D1,..., Dk, D1,..., Dk] trong đó:
output[i1,..., ik, i1,..., ik] = diagonal[i1, ..., ik] và 0 ở mọi nơi khác.
Ví dụ:
# 'diagonal' is [1, 2, 3, 4]
tf.diag(diagonal) ==> [[1, 0, 0, 0]
[0, 2, 0, 0]
[0, 0, 3, 0]
[0, 0, 0, 4]]Luận điểm:
- phạm vi: Một đối tượng Phạm vi
- Đường chéo: Tensor hạng k, trong đó k nhỏ hơn hoặc bằng 1.
Trả về:
-
Output: Tensor đầu ra.
Hàm khởi tạo và hàm hủy | |
|---|---|
Diag (const :: tensorflow::Scope & scope, :: tensorflow::Input diagonal) |
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 | |
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
Chẩn đoán
Diag( const ::tensorflow::Scope & scope, ::tensorflow::Input diagonal )
nút
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const