Transforms

ToTensor

class ToTensor[source]

Convert ndarrays in sample to Tensors.

Normalize

class Normalize(mean, std, unravel_model)[source]

Normalize samples.

Compose

class Compose(transforms)[source]

Composes several transforms together.