• 차원이 없는 값: 스칼라
  • 1D: Vector
  • 2D: Matrix
  • 3D: Tensor

 

  • 2D Tensor |t| = (batch_size, dim)
  • batch_size: 세로
  • dim: 가로

 

  • 3D Tensor |t| = (batch_size, width = length, height = dim)
  • batch_size: 세로
  • width = length: 가로
  • height = dim: 깊이

 


  • torch.FloatTensor: 차원 생성
  • t.dim(): 차원수
  • t.shape(): 차원의 형태

 

2D FloatTensor

 


FloatTensor 연산

마지막 부분의

m1 = 1x2는 [[1, 2], [1, 2]]로,

m2 = 2x1은 [[3, 3], [4, 4]]로,

바뀐다.

 


  • dim = 0: 행 기준
  • dim = 1: 열 기준

 

 

 


출처:

https://www.youtube.com/watch?v=St7EhvnFi6c&list=PLQ28Nx3M4JrhkqBVIXg-i5_CVVoS1UzAv&index=2 

 

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

07_Pytorch_Logistic_Regression  (0) 2021.08.18
05_Pytorch_nn.Module  (0) 2021.08.13
04_Pytorch_Multivariate_Linear_Regression  (0) 2021.08.12
03_Pytorch_Linear_Regression  (0) 2021.08.12
02_Pytorch Shape and Calculation  (0) 2021.08.10

+ Recent posts