def expand_dims(a: BlockArray, axis): return a.expand_dims(axis)
def expand_dims(x: BlockArray, axis): return x.expand_dims(axis)