TensorDiagPart

lớp cuối cùng công khai TensorDiagPart

Trả về phần đường chéo của tensor.

Thao tác này trả về một tenxơ có phần `đường chéo` của `đầu vào`. Phần `chéo` được tính như sau:

Giả sử `input` có kích thước `[D1,..., Dk, D1,..., Dk]`, thì đầu ra là một tensor hạng `k` có kích thước `[D1,..., Dk]` Ở đâu:

`đường chéo[i1,..., ik] = input[i1, ..., ik, i1,..., ik]`.

Ví dụ:

# 'input' is [[1, 0, 0, 0]
               [0, 2, 0, 0]
               [0, 0, 3, 0]
               [0, 0, 0, 4]]
 
 tf.diag_part(input) ==> [1, 2, 3, 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 > TensorDiagPart <T>
tạo (Phạm vi phạm vi, đầu và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 TensorDiagPart mới.
Đầu ra <T>
đường chéo ()
Đường chéo được trích xuất.

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

Giá trị không đổi: "DiagPart"

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 TensorDiagPart <T> tạo (Phạm vi phạm vi, đầu và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 TensorDiagPart mới.

Thông số
phạm vi phạm vi hiện tại
đầu vào Xếp hạng tenxơ k trong đó k chẵn và không bằng 0.
Trả lại
  • một phiên bản mới của TensorDiagPart

Đầu ra công khai <T> đường chéo ()

Đường chéo được trích xuất.