Change the get_shape virtual function in base_dtype