代碼example:
import torch
import torch.nn as nn# 定義一些增強操作,例如隨機水平翻轉和歸一化
augmentations = nn.Sequential(nn.RandomHorizontalFlip(),nn.Normalize(mean=[0.5], std=[0.5])
)# 創建一個示例 tensor
candidate = torch.randn(1, 3, 224, 224) # 假設這是一個圖像 batch# 應用增強操作
candidate_augmented = candidate.clone()
candidate_augmented.data = augmentations(candidate.data)