线性变换

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

使用离线计算的正方形变换矩阵和mean_vector转换张量图像或视频。

此转换不支持PIL图像。 给定transformation_matrix和mean_vector，将展平torch.*Tensor并从中减去mean_vector，然后计算与转换矩阵的点积，最后将张量重塑为其原始形状。

Applications:

白化变换：假设X是一个零均值的列向量数据。 然后使用torch.mm(X.t(), X)计算数据的协方差矩阵[D x D]， 对该矩阵进行SVD分解并将其作为变换矩阵传递。

Parameters:

• transformation_matrix (张量) – 张量 [D x D], D = C x H x W

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