torch.diagonal_scatter¶
- torch.diagonal_scatter(input, src, offset=0, dim1=0, dim2=1) Tensor ¶
将
src
张量的值嵌入到input
中,沿input
的对角元素,相对于dim1
和dim2
。此函数返回一个具有新存储的张量;它不返回视图。
参数
offset
控制要考虑的对角线如果
offset
= 0,则它是主对角线。如果
offset
> 0,则它位于主对角线上方。如果
offset
< 0,则它位于主对角线下方。
- 参数
注意
src
必须具有适当的大小才能嵌入到input
中。具体来说,它应该与torch.diagonal(input, offset, dim1, dim2)
的形状相同示例
>>> a = torch.zeros(3, 3) >>> a tensor([[0., 0., 0.], [0., 0., 0.], [0., 0., 0.]]) >>> torch.diagonal_scatter(a, torch.ones(3), 0) tensor([[1., 0., 0.], [0., 1., 0.], [0., 0., 1.]]) >>> torch.diagonal_scatter(a, torch.ones(2), 1) tensor([[0., 1., 0.], [0., 0., 1.], [0., 0., 0.]])