잡다한 삽질 스튜디오

2024년 3월 16일

/

PyTorch, 1인 1프로젝트

PyTorch: 텐서(Tensor)와 차원

PyTorch: 텐서(Tensor)와 차원

텐서(Tensor) 란?

텐서(tensor)는 데이터 표현을 위한 기본 구조로, 데이터를 담기위한 컨테이너(container)로서 일반적으로 수치형 데이터를 저장한다. 넘파이(NumPy)의 ndarray와 유사하고, GPU를 사용한 연산 가속이 가능하다.

텐서 차원

이수안컴퓨터연구소

0D Tensor (Scalar)

Scalar는 숫자 하나라고 생각하면 된다.

1D Tensor (Vector)

Vector는 1차원 배열이라고 생각하면 된다.

2D Tensor (Matrix)

Matrix는 행렬로 행과 열을 가지는 2차원 형상을 가지고 있는 구조다.

\(|t| = (batch size, dim)\)

3D Tensor (Tensor)

Tensor는 다차원 배열이라고 부른다.

  • 이미지

    • \(|t| = (batch size, width, height)\)
  • 동영상

    • \(|t| = (batch size, length, dim)\)
/emoji//end/

참고

PyTorch로 시작하는 딥 러닝 입문
이 책은 딥 러닝 프레임워크 PyTorch를 사용하여 딥 러닝에 입문하는 것을 목표로 합니다. 이 책은 파이썬은 어느정도 할 줄 안다고 가정합니다. 이 이상의 …