LinearTransformation

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

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

Warning

The LinearTransformation transform is in Beta stage, and while we do not expect major breaking changes, some APIs may still change according to user feedback. Please submit any feedback you may have in this issue: https://github.com/pytorch/vision/issues/6753, and you can also check out https://github.com/pytorch/vision/issues/7319 to learn more about the APIs that we suspect might involve future changes.

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.

Applications:

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

Parameters:

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

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