# from typing import Union
# import numpy as np
# import torch

# # For input arrays
# ArrayLike = Union[
#     np.ndarray[Union[np.float64, np.complex128], np.dtype[Union[np.float64, np.complex128]]], torch.Tensor
# ]

# # For output arrays
# OutputArray = Union[
#     np.ndarray[Union[np.float64, np.complex128], np.dtype[Union[np.float64, np.complex128]]], torch.Tensor
# ]
