在Torch中,生成對抗網絡(GAN)模塊有以下幾個常用的庫:
nn.Sequential
:用于構建生成器和判別器網絡的序列模塊。nn.Linear
:用于定義全連接層。nn.Conv2d
:用于定義卷積層。nn.BatchNorm2d
:用于定義批歸一化層。nn.ReLU
:用于定義激活函數。nn.LeakyReLU
:用于定義帶泄漏的激活函數。nn.Sigmoid
:用于定義輸出層的激活函數。nn.BCELoss
:用于定義二進制交叉熵損失函數。nn.L1Loss
:用于定義L1損失函數。nn.MSELoss
:用于定義均方誤差損失函數。這些模塊可以組合在一起來構建生成對抗網絡,通過訓練生成器和判別器來實現生成器生成接近真實數據的樣本,以欺騙判別器。