expand_dims

import numpy as np

x = np.array([3,4])
x.shape
>> (2,0)
y = np.expand_dims(x, axis = 0)
y
>> array([[3,4]])
y.shape
>> (1,2)

expand_dims 는 배열에 축을 추가하는 작업을 해주는 함수이다.
아래 그림을 보면 이해가 쉽다.

즉, expand_dims(array, axis) 의 axis 는 어떤 축을 추가할지를 지정해주는 변수이다.

[ref]
https://webnautes.tistory.com/1529