pytorch ツール - pytorch の autograd

torch.tensor について

ここに画像の説明を挿入

テンソルの演算について

x1=torch.ones(3,3)
x=torch.ones(2,2,requires_grad=True)
print(x1,'\n',x)

ここに画像の説明を挿入

y=x+2
print(y)
print(x.grad_fn)
print(y.grad_fn)

ここに画像の説明を挿入

z=y*y*3
out=z.mean()
print(z,out)

ここに画像の説明を挿入

知らせ
ここに画像の説明を挿入

a=torch.randn(2,2)
a=((a*3)/(a-1))
print(a.requires_grad)
a.requires_grad_(True)
print(a.requires_grad)
b=(a*a).sum()
print(b)
print(b.grad_fn)

ここに画像の説明を挿入

グラデーションについて

out.backward()
print(x.grad)

ここに画像の説明を挿入

print(x.requires_grad)
print((x**2).requires_grad)
with torch.no_grad():
    print((x**2).requires_grad)

ここに画像の説明を挿入

y=x.detach()
print(y.requires_grad)

print(x)
print(y)
print(x.eq(y).all())

ここに画像の説明を挿入

おすすめ

転載: blog.csdn.net/qq_40527560/article/details/131866978