파이토치1 예제로 배우는 파이토치3. 마지막. 파이토치 튜토리얼 3번째~!!! 이 글은 파이토치 공식 튜토리얼에서 가져온 예제다. 더 정확한 내용을 보시려면 여기를 참조 사용자 정의 nn.Module 때때로 기존 모듈의 구성보다 더 복잡한 모델을 구성해야할 때가 있습니다. 이럴 때는 nn.Module의 서브클래스로 새 모듈을 정의하고 입력 tnesor를 받아 다른 모듈 또는 tensor의 autograd 연산을 사용하여 출력 Tensor를 만드는 forward를 정의 해보자. 이전 포스팅에서 만들었던 모델은 nn.Sequential 모델이었다면 이제는 사용자정의 모델이다. import torch class TwoLayerNet(torch.nn.Module): def __init__(self, D_in, H, D_out): """ 생성자에서 2개의 n.. 2020. 8. 25. 이전 1 다음