k_expand_dims function

Adds a 1-sized dimension at index axis.