pytorch-Day0
ai·2025. 7. 9.
1.1번만 돌리는 ai 흐름1)데이터x,y입력2)w초기화3)예측값4)손실함수5)역전파 -> 기울기계산6.w초기화 -> w업데이트(기울기 계산한거 바탕으로) 코드import torch#1.x,y 입력x = torch.tensor([1.0,2.0,3.0])y_true = torch.tensor([2.0,4.0,6.0])#2.w 초기화w = torch.tensor([0.1],requires_grad=True)#3.예측값 만들기 w * xy_pred = w * x#4.손실값 계산loss = torch.mean((y_pred - y_true) ** 2)#(w,예측값,손실값 출력)print(f"현재 w: {w.item():.4f}")print(f"예측값:{y_pred}")print(f"손실값:{loss.item()..