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

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

• Output : Tenxor đầu ra.

Thuộc tính công khai

Operation operation

::tensorflow::Output output

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const