PyTorch Autograd를 이용한 자동 미분 구현
신경망 학습에서 가장 널리 사용되는 역전파 알고리즘은 매개변수(가중치)를 손실 함수의 기울기에 따라 조정합니다. PyTorch는 이러한 기울기 계산을 위해 torch.autograd 자동 미분 엔진을 제공하며, 임의의 계산 그래프에 대한 미분을 지원합니다.
계산 그래프 구성 예제
import torch
inputs = torch.ones(4) # 입력 텐서
targets = torch.zeros(2) # 예측 ...
6월 9일 23:47에 게시됨