Replace diagonal of matrix with new diagonal values

1

Numpy Version of the Code:

image

For Pytorch version, It seems like we cannot replace diagonal elements with a vector of elements and only a scalar value can be filled on diagonal.

image
image745×357 27.7 KB

2

This older post provides some approaches using direct indexing to fill the diagonal with a tensor.

1 Like
3

Thank you. It worked. I should have checked it better.