dòng chảy căng:: ôi:: Đường chéo

#include <array_ops.h>

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

Bản tóm tắt

Cho một diagonal , thao tác này trả về một tenxơ có 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ó các kích thước [D1,..., Dk] thì đầu ra là một tenxơ 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]]

Lập luận:

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

Trả về:

Hàm 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

Đường chéo

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

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