LinearTransformation

class torchvision.transforms.LinearTransformation(transformation_matrix, mean_vector)

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

應用

白化變換：假設 X 是一個零均值的列向量資料。然後計算資料協方差矩陣 [D x D]，使用 torch.mm(X.t(), X)，對該矩陣執行 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) – 要進行白化的張量影像。

返回:

轉換後的影像。

返回型別:

張量