torch.Tensor.diagonal_scatter

Tensor.diagonal_scatter(src, offset=0, dim1=0, dim2=1) → Tensor

See torch.diagonal_scatter()