PyTorch torch.diagonal_scatter 関数

Pytorch torch リファレンスマニュアル

torch.diagonal_scatterは、PyTorchにおいて値をテンソルの対角線位置に散布するための関数です。これはsrcの値をinputの指定された対角線上に散布します。

関数定義

torch.diagonal_scatter(input, src, offset=0, dim1=0, dim2=1)

引数:

• input(Tensor): 入力テンソル、すなわち修正するテンソルです。
• src(Tensor): ソーステンソル、対角線位置に散布する値です。
• offset(int, オプション): 対角線のオフセット量。正の値は上対角線、負の値は下対角線、0は主対角線を表します。
• dim1(int, オプション): 最初の次元。デフォルトは0です。
• dim2(int, オプション): 2番目の次元。デフォルトは1です。

戻り値:

• torch.Tensor: 修正されたテンソルを返します。

使用例

出力結果は次のとおりです：

输入张量:
tensor([[0., 0., 0., 0.],
        [0., 0., 0., 0.],
        [0., 0., 0., 0.],
        [0., 0., 0., 0.]])

源张量:
tensor([1, 2, 3, 4])

散布到主对角线:
tensor([[1., 0., 0., 0.],
        [0., 2., 0., 0.],
        [0., 0., 3., 0.],
        [0., 0., 0., 4.]])

出力結果は次のとおりです：

散布到上对角线 (offset=1):
tensor([[0., 1., 0., 0.],
        [0., 0., 2., 0.],
        [0., 0., 0., 3.],
        [0., 0., 0., 0.]])

散布到下对角线 (offset=-1):
tensor([[0., 0., 0., 0.],
        [1., 0., 0., 0.],
        [0., 2., 0., 0.],
        [0., 0., 3., 0.]])

出力結果は次のとおりです：

输入形状: torch.Size([3, 4, 4])
源形状: torch.Size([3])
结果形状: torch.Size([3, 4, 4])

第一个batch的结果:
tensor([[10., 0., 0., 0.],
        [0., 20., 0., 0.],
        [0., 0., 30., 0.],
        [0., 0., 0., 0.]])

注意：torch.diagonal_scatter元の入力テンソルを変更せず、代わりに新しいテンソルを返します。srcのサイズは対角線の要素数と一致する必要があります。

Pytorch torch リファレンスマニュアル