PyTorch & Tensor

PyTorch 란

PyTorch는 오픈 소스 딥러닝 프레임워크로, 주로 인공지능과 머신러닝 작업에 사용됩니다. 2016년에 페이스북의 인공지능 연구팀에 의해 처음 개발되었으며, 연구와 생산 환경에서 모두 널리 사용됩니다.

Tensor

다차원 배열을 일반화한 개념으로, 딥러닝과 머신러닝에서 데이터를 표현하는 데 사용

• 0-D Tensor (Scalar)

  # 0-D Tensor 생성
  scalar = torch.tensor(5)
  print(scalar) # tensor(5)

• 1-D Tensor (Vector)

  # 1-D Tensor 생성
  vector = torch.tensor([1, 2, 3])
  print(vector) # tensor([1, 2, 3])

• 2-D Tensor (Matrix)

  # 2-D Tensor 생성
  matrix = torch.tensor([[1, 2, 3], [4, 5, 6]])
  print(matrix)
  
  # tensor([[1, 2, 3],
  #         [4, 5, 6]])

• 3-D Tensor 3차원 Tensor은 색(RGB)의 표현등에 사용된다.

  # 3-D Tensor 생성
  T = torch.tensor([[[1, 2, 3], [4, 5, 6]],
                         [[7, 8, 9], [10, 11, 12]]])
  print(T)
  # tensor([[[ 1,  2,  3],
  #          [ 4,  5,  6]],
  # 
  #         [[ 7,  8,  9],
  #          [10, 11, 12]]])

• N-D Tensor