
[PyTorch] torch.nn
·
ML & DL/PyTorch
torch.nn PyTorch에서 신경망 모델을 구현하는 데 필요한 여러 가지 클래스와 함수를 제공해주는 모듈입니다. nn.Linear 선형 변환(Linear Transformation)을 수행하는 클래스입니다. nn.Linear를 사용하여 tensor의 크기 or 모양을 반환합니다. # 입력 텐서의 차원은 5이고, 출력 텐서의 차원은 3인 nn.Linear 모듈을 생성합니다. linear_layer = nn.Linear(5, 3) # 입력 텐서를 생성합니다. input_tensor = torch.randn((2, 5)) # nn.Linear 모듈을 통해 입력 텐서를 출력 텐서로 변환합니다. output_tensor = linear_layer(input_tensor) print(output_tensor...