Study/Deep Learning
Pytorch 00. Hello Pytorch
Rylah
2022. 2. 11. 09:26
import torch
# 입력 뉴런 : 5개, 출력 뉴런 : 3개
# 총 사용 시냅스 15개
hello = torch.nn.Linear(5, 3)
# 임의의 입력 벡터 2개 만들기
data = torch.randn(2, 5)
# 인공 신명망에 입력하고 결과를 출력
print(data)
print(hello(data))