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ề:

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