tensorflow::ops::MatrixDiagPart

Trả về phần đường chéo được nhóm lại của một tensor được nhóm lại.

Bản tóm tắt

Thao tác này trả về một tensor chứa phần diagonal của dữ input được nhóm lại. Phần diagonal được tính như sau:

Giả sử input có k chiều [I, J, K, ..., M, N] , thì đầu ra là một tensor hạng k - 1 với các chiều [I, J, K, ..., min(M, N)] trong đó:

diagonal[i, j, k, ..., n] = input[i, j, k, ..., n, n] .

Dữ liệu đầu vào phải ít nhất là một ma trận.

Ví dụ:

# 'input' is [[[1, 0, 0, 0]
               [0, 2, 0, 0]
               [0, 0, 3, 0]
               [0, 0, 0, 4]],
              [[5, 0, 0, 0]
               [0, 6, 0, 0]
               [0, 0, 7, 0]
               [0, 0, 0, 8]]]

and input.shape = (2, 4, 4)

tf.matrix_diag_part(input) ==> [[1, 2, 3, 4], [5, 6, 7, 8]]

which has shape (2, 4)

Luận điểm:

• phạm vi: Một đối tượng Phạm vi
• Đầu vào: Tensor hạng k với k >= 2 .

Trả về:

• Output : Đường chéo được trích xuất có hình dạng diagonal.shape = input.shape[:-2] + [min(input.shape[-2:])] .

Thuộc tính công khai

::tensorflow::Output diagonal

Operation operation

 MatrixDiagPart(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const