Source code for torch_brain.utils.misc

from packaging import version
import numpy as np




def np_string_prefix(prefix: str, array: np.ndarray) -> np.ndarray:
    """
    Adds a string prefix to each element of a numpy string array.

    Args:
        prefix (str): The string to prepend to each element.
        array (np.ndarray): An array of strings or string-like objects.

    Returns:
        np.ndarray: New array with the prefix added to each element.
    """
    if version.parse(np.__version__) >= version.parse("2.0"):
        return np.strings.add(prefix, array)
    else:
        return np.core.defchararray.add(prefix, array)