PyTorch torch.diagflat 関数
Pytorch torch リファレンスマニュアルtorch.diagflatPyTorch で対角フラット行列を作成するための関数です。入力が1次元でも多次元でも、フラット化されて対角要素になります。
関数定義
torch.diagflat(input, diagonal=0)
パラメータ説明：
input: 入力テンソル。フラット化されます。diagonal: 対角インデックス
使用例
例
import torch
# 1次元テンソルを作成
x = torch.tensor([1, 2, 3])
# 対角フラット行列を作成
y = torch.diagflat(x)
print(y)
# 1次元テンソルを作成
x = torch.tensor([1, 2, 3])
# 対角フラット行列を作成
y = torch.diagflat(x)
print(y)
出力結果は：
tensor([[1, 0, 0],
[0, 2, 0],
[0, 0, 3]])
例
import torch
# 2次元テンソルを作成
x = torch.tensor([[1, 2], [3, 4]])
# [1,2,3,4] にフラット化され、4x4 の対角行列を作成
y = torch.diagflat(x)
print(y)
# 2次元テンソルを作成
x = torch.tensor([[1, 2], [3, 4]])
# [1,2,3,4] にフラット化され、4x4 の対角行列を作成
y = torch.diagflat(x)
print(y)
出力結果は：
tensor([[1, 0, 0, 0],
[0, 2, 0, 0],
[0, 0, 3, 0],
[0, 0, 0, 4]])
その他の拡張