wildpieces

  • 홈
  • 태그
  • 방명록

pytorch 1

nn.Linear

y = Wx + bnn.Linear(in_features, out_features)x: 입력 벡터 (크기: in_features)W: 가중치 행렬 (크기: out_features * in_features)b: 편향 벡터 (크기: out_features)y: 출력 벡터 (크기: out_features)=> 입력 벡터 x를 선형 변환해서 출력 벡터 y를 만들어주는 레이어사용예시import torchimport torch.nn as nnlinear = nn.Linear(in_features=3, out_features=2)x = torch.tensor([[1.0, 2.0, 3.0]]) # shape: (1, 3)y = linear(x) # shape: (1, 2)print(y)3차원을 입력 받아서 2차원으로 ..

Deep learning 2025.04.11
이전
1
다음
더보기
프로필사진

wildpieces

https://github.com/wildpieces

  • 분류 전체보기 (8)
    • Deep learning (4)
    • Agentic AI (0)
    • Dev (1)
    • AWS (2)
    • Lecture (1)

Tag

Adam, 가상환경, residual connection, chunk, vanishing gradient, pooling, SGD, dl, deeplearning, langchain, bedrock, pytorch, aws, deep learning, pyenv, dropout, Python, virtualenv, Optimizer, Activation Function,

Copyright © Kakao Corp. All rights reserved.

티스토리툴바