class torchvision.transforms.v2.LinearTransformation(transformation_matrix: Tensor, mean_vector: Tensor)[source]

[BETA] Transform a tensor image or video with a square transformation matrix and a mean_vector computed offline.


The LinearTransformation transform is in Beta stage, and while we do not expect disruptive breaking changes, some APIs may slightly change according to user feedback. Please submit any feedback you may have in this issue:

This transform does not support PIL Image. Given transformation_matrix and mean_vector, will flatten the torch.*Tensor and subtract mean_vector from it which is then followed by computing the dot product with the transformation matrix and then reshaping the tensor to its original shape.


whitening transformation: Suppose X is a column vector zero-centered data. Then compute the data covariance matrix [D x D] with, X), perform SVD on this matrix and pass it as transformation_matrix.

  • transformation_matrix (Tensor) – tensor [D x D], D = C x H x W

  • mean_vector (Tensor) – tensor [D], D = C x H x W


Access comprehensive developer documentation for PyTorch

View Docs


Get in-depth tutorials for beginners and advanced developers

View Tutorials


Find development resources and get your questions answered

View Resources