Related to: Machine Learning

torch.tensor의 requires_grad param의 기능

  • requires_grad = True일 경우, 해당 tensor는 model이 backward 될 때 자동미분(auto grad)을 수행합니다.

참조

Week 2