Rylah's Study & Daily Life

Pytorch 00. Hello Pytorch 본문

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

 

'Study > Deep Learning' 카테고리의 다른 글

PyTorch 02. MNIST  (0) 2022.02.11
PyTorch 01. Basic PyTorch  (0) 2022.02.11
[Tensorflow] 04. RNN - AirLine  (0) 2022.01.26
[TensorFlow] 03. CNN - Fashion Items  (0) 2022.01.26
[TensorFlow] 02. DNN - Boston House Value  (0) 2022.01.23