dosma.core.numpy_routines.expand_dims

dosma.core.numpy_routines.expand_dims(x, axis: Union[int, Sequence[int]])[source]

Expand across non-spatial dimensions.

Parameters:

x (MedicalVolume) – A medical image.
axis (int(s)) – Axis/axes to expand dimensions.

Returns:

The medical image with expanded dimensions.

Return type:

MedicalVolume

Examples

>>> mv = dm.MedicalVolume(np.random.rand(3,4,5), affine=np.eye(4))
>>> mv_expanded = np.expand_dims(mv, axis=-1)  # Expand last dimension
>>> mv_expanded.shape
(3, 4, 5, 1)