dòng chảy căng:: ôi:: Ma TrậnDiagPhần

Trả về phần đường chéo theo đợt của một tenxơ theo đợt.

Bản tóm tắt

Hoạt động này trả về một tensor với phần diagonal của input đợt. 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 có kích thước [I, J, K, ..., min(M, N)] Ở đâu:

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

Đầu vào ít nhất phải là 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)

Lập luận:

• phạm vi: Một đối tượng Phạm vi
• đầu vào: Xếp hạng k tensor trong đó k >= 2 .

Trả về:

• Output : (Các) đườ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