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 cộng
Phương pháp kế thừa
boolean | bằng (Đối tượng arg0) |
Lớp cuối cùng<?> | getClass () |
int | mã băm () |
khoảng trống cuối cùng | thông báo () |
khoảng trống cuối cùng | thông báoTất cả () |
Sợi dây | toString () |
khoảng trống cuối cùng | chờ đã (arg0 dài, int arg1) |
khoảng trống cuối cùng | chờ đã (arg0 dài) |
khoảng trống cuối cùng | Chờ đợi () |
môi trường thực thi trừu tượng | env () Trả về môi trường thực thi mà op này được tạo trong đó. |
hoạt động trừu tượng |
Đầu ra trừu tượng <T> | asOutput () Trả về tay cầm tượng trưng của tensor. |
trừu tượng T | asTensor () Trả về tensor ở toán hạng này. |
hình dạng trừu tượng | |
Lớp trừu tượng<T> | kiểu () Trả về kiểu tensor của toán hạng này |
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 cộng
Đầ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