emmi.modules.preprocessors.numpy_to_torch_tensor¶

Classes¶

Convert numpy arrays to PyTorch tensors.

class emmi.modules.preprocessors.numpy_to_torch_tensor.NumpyToTorchTensorPreProcessor(normalization_key)¶

Bases: ksuit.data.preprocessors.PreProcessor

Convert numpy arrays to PyTorch tensors.

Parameters:: normalization_key (str) – key to identify on which getitem_ in the dataset/tensor the preprocessor is applied.
Raises:: TypeError – If normalization_key is not a string.

denormalize(x)¶

Convert a PyTorch tensor back to a numpy array.