[numpy, tensorflow] numpy.expand_dims , tensorflow.expand_dims

numpy.expand_dims

Paramaters

a : array_like

axis : tuple, int, ints ...  ( axis > a.ndim -> axis == a.ndim, axis < -a.ndim - 1 -> axis == 0 | version 1.13.0 이후)

Return : numpy array

 

- input 으로 리스트 뿐만 아니라 상수도 가능함. 하지만 axis=0, -1 이 아니면 오류발생

 

output_array = np.expand_dims(image, axis=(0,1))

 

tensorflow.expand_dims

Parameters

input : Tensor

axis : integer ( -(D+1) <= axis <= D ) , 리스트 안됨

name : output Tensor 의 이름 (Optional)

Return : Tensor

 

- input 으로 array 혹은 ndarray 도 가능함.

 

output_tensor = tf.expand_dims(image, axis=0)