Trả về một tenxơ đường chéo với các giá trị đường chéo cho trước.
Với một `đường chéo`, thao tác này trả về một tenxơ có `đường chéo` và mọi thứ khác được đệm bằng số không. Đường chéo được tính như sau:
Giả sử `đường chéo` có các kích thước [D1,..., Dk], thì đầu ra là một tensor hạng 2k với các kích thước [D1,..., Dk, D1,..., Dk] trong đó:
`output[i1,..., ik, i1,..., ik] = chéo[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]]
Hằng số
| Sợi dây | OP_NAME | Tên của op này, được biết đến bởi công cụ lõi TensorFlow |
Phương pháp công khai
| Đầu ra <T> | asOutput () Trả về tay cầm tượng trưng của tensor. |
| tĩnh <T mở rộng TType > TensorDiag <T> | |
| Đầu ra <T> | đầu ra () |
Phương pháp kế thừa
Hằng số
Chuỗi cuối cùng tĩnh công khai OP_NAME
Tên của op này, được biết đến bởi công cụ lõi TensorFlow
Phương pháp công khai
Đầu ra công khai <T> asOutput ()
Trả về tay cầm tượng trưng của tensor.
Đầu vào của các hoạt động TensorFlow là đầu ra của một hoạt động TensorFlow khác. Phương pháp này được sử dụng để thu được một thẻ điều khiển mang tính biểu tượng đại diện cho việc tính toán đầu vào.
public static TensorDiag <T> tạo ( Phạm vi phạm vi, Đường chéo toán hạng <T>)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác TensorDiag mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| đường chéo | Xếp hạng tenxơ k trong đó k lớn nhất là 1. |
Trả lại
- một phiên bản mới của TensorDiag