dòng chảy :: hoạt động :: Chẩn đoán

#include <array_ops.h>

Trả về một tenxơ theo đường chéo với các giá trị đường chéo đã cho.

Tóm lược

Cho một diagonal , phép toán này trả về một tensor với diagonal và mọi thứ khác được đệm bằng số không. Đường chéo được tính như sau:

Giả sử diagonal có kích thước [D1, ..., Dk], khi đó đầu ra là một tenxơ bậc 2k với cá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]]

Tranh luận:

  • phạm vi: Một đối tượng Phạm vi
  • đường chéo: Xếp hạng k tensor trong đó k nhiều nhất là 1.

Lợi nhuận:

  • Output : Bộ căng đầu ra.

Người xây dựng và Người phá hủy

Diag (const :: tensorflow::Scope & scope, :: tensorflow::Input diagonal)

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

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

Chẩn đoán

 Diag(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input diagonal
)

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