tensorflow::ops::MatrixSetDiag

Trả về một tensor ma trận được nhóm lại với các giá trị đường chéo được nhóm lại mới.

Bản tóm tắt

Với input và diagonal , phép toán này trả về một tensor có cùng hình dạng và giá trị với input , ngoại trừ đường chéo chính của các ma trận trong cùng. Các giá trị này sẽ bị ghi đè bởi các giá trị trong diagonal .

Kết quả được tính toán như sau:

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

• output[i, j, k, ..., m, n] = diagonal[i, j, k, ..., n] for m == n .
• output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] với điều kiện m != n .

Luận điểm:

• phạm vi: Một đối tượng Phạm vi
• Đầu vào: Hạng k+1 , trong đó k >= 1 .
• Đường chéo: Hạng k , trong đó k >= 1 .

Trả về:

• Output : Hạng k+1 , với output.shape = input.shape .

Thuộc tính công khai

Operation operation

::tensorflow::Output output

 MatrixSetDiag(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input diagonal
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const