DataModule.normalize_x

DataModule.normalize_x(x: torch.Tensor | ndarray) → torch.Tensor | ndarray

Normalize only the input data. Takes care of the case where x is a numpy array or a torch tensor.

Parameters:

x (Union[torch.Tensor, np.ndarray]) – A numpy array or a torch tensor to be normalized.

Returns:

Union[torch.Tensor, np.ndarray] – The normalized data. The output has the same type as the input.