728x90 backward1 [PyTorch] torch.nn.Module에 대해서 torch.nn.Module - 딥러닝을 구성하는 Layer의 base class - Input, Output, Forward, Backward 정의 - 학습의 대상이 되는 parameter(tensor) 정의 파이토치 공식 document에서 명시한 코드로 예시를 써놓는다. https://pytorch.org/docs/stable/generated/torch.nn.Module.html#torch.nn.Module Module — PyTorch 1.13 documentation Shortcuts pytorch.org import torch.nn as nn # 신경망에서 일반적으로 사용되는 모든 연산은 torch.nn모듈에서 활용 가능 import torch.nn.functional as F class M.. 2022. 12. 24. 이전 1 다음 728x90
