LinearTransformation

class torchvision.transforms.LinearTransformation(transformation_matrix, mean_vector)

使用離線計算的平方轉換矩陣和 mean_vector 轉換張量圖像。此轉換不支援 PIL Image。給定 transformation_matrix 和 mean_vector，將展平 torch.*Tensor 並从中減去 mean_vector，然後計算與轉換矩陣的點積，然後將張量重塑為其原始形狀。

應用

白化轉換：假設 X 是一個以零為中心資料的列向量。然後使用 torch.mm(X.t(), X) 計算資料共變異數矩陣 [D x D]，對該矩陣執行 SVD 並將其作為 transformation_matrix 傳遞。

參數:

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

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

forward(tensor: Tensor) → Tensor

參數:

tensor (Tensor) – 要白化的張量圖像。

傳回:

轉換後的圖像。

傳回類型:

Tensor