dòng chảy căng:: ôi:: MatrixSetDiag

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

Bản tóm tắt

Cho input và diagonal , thao tác này trả về một tenxơ có hình dạng và giá trị giống như input , ngoại trừ đường chéo chính của ma trận trong cùng. Chúng sẽ bị ghi đè bởi các giá trị theo diagonal .

Đầu ra được tính 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 có kích thước [I, J, K, ..., M, N] trong đó:

• output[i, j, k, ..., m, n] = diagonal[i, j, k, ..., n] cho m == n .
• output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] cho m != n .

Lập luận:

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

Trả về:

• Output : Xếp 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