get_shape function

get_shape