reshape, flatten

reshape

배열의 구조 변경하는 method로, 행이나 열(혹은 특정 차원)을 늘리거나 줄임

단, 전체 데이터 개수는 유지되어야함

# reshape한 데이터 개수 != 원본 데이터 개수
x = np.array(range(8))
x.reshape(2,2)

>ValueError

 

-1 : 전체 요소의 개수는 고정시키고 -1을 쓴 자리를 제외한 나머지 차원의 크기를 지정했을때 전체 요소의 개수를 고려하여 -1자리의 차원이 자동으로 지정됨

x.reshape(2,-1)
#2행 지정, 열은 자동 계산

>array([[0, 1, 2, 3],

          [4, 5, 6, 7]])

---

flatten()

어떤 tensor든지 1차원 벡터로 변환 (= 평탄화)
이때 데이터 개수와 값은 그대로 유지됨

x.flatten()

>array([0, 1, 2, 3, 4, 5, 6, 7])