Expand the shape of an array by inserting a new axis, corresponding to
a given position in the array shape. Useful when predicting a model based
on a single input.
Usage
expand_dims(a, axis = 0)
Arguments
a
array to expand
axis
position (amongst axes) where new axis is to be inserted.