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))