→ 랜덤 생성한 텐서의 값을 경사하강법을 활용해 이리저리 바꿔가며 함수의 출력값(목표텐서와의 차이 값)을 최소화하도록 하기
import torch
import torch.nn.functional as F
target = torch.FloatTensor([[.1,.2,.3,],
[.4,.5,.6],
[.7,.8,.9]])
x = torch.rand_like(target)
x.requires_grad = True
print(x)
결과
tensor([[0.8888, 0.7825, 0.4902],
[0.4046, 0.9947, 0.0120],
[0.6554, 0.3357, 0.7663]], requires_grad=True)
loss = F.mse_loss(x, target)
loss
결과
tensor(0.2026, grad_fn=<MseLossBackward0>)