I'm not sure about the name either. I'm wary of using ```expand_dims``` though because the api is so different i.e. you don't specify the axes.