线性变换

class torchvision.transforms.LinearTransformation(transformation_matrix, mean_vector)

使用离线计算的正方形变换矩阵和均值向量对张量图像进行变换。 此变换不支持PIL图像。 给定变换矩阵和均值向量，将展平torch.*Tensor并从中减去均值向量，然后计算与变换矩阵的点积，最后将张量重塑为其原始形状。

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

forward(tensor: Tensor) → Tensor

Parameters:

tensor (Tensor) – 需要白化的张量图像。

Returns:

转换后的图像。

Return type:

张量