Pytorch: Suara gumbel

Hai,

Saya sudah mencoba menambahkan suara Gumbel seperti di sini , tetapi tidak berhasil.
Sepertinya beberapa modul masih hilang (misalnya, fungsi variabel dasar seperti nn.Uniform()), atau apakah saya salah? Bagaimana Anda menerapkan di pytorch misalnya baris berikut:

``` -- Buat noise modul sampel
noiseModule lokal = nn.Sequential()
no iseModule:add (nn.Uniform(0, 1)) -- Contoh dari U(0, 1)
-- Ubah sampel seragam menjadi sampel Gumbel
no iseModule:add (nn.AddConstant(1e-9, true)) -- Meningkatkan stabilitas numerik
tidak ada iseModule:tambahkan (nn.Log())
tidak ada iseModule:add (nn.MulConstant(-1, true))
no iseModule:add (nn.AddConstant(1e-9, true)) -- Meningkatkan stabilitas numerik
tidak ada iseModule:tambahkan (nn.Log())
tidak ada iseModule:add (nn.MulConstant(-1, true))

-- Buat sampler q(z) = G(z) = softmax((log(π) + )/τ) (trik reparametrisasi)
sampler lokal = nn.Sequential()
samplerInternal lokal = nn.ConcatTable()
sample rInternal:add (nn.Identity()) -- log probabilitas yang tidak dinormalisasi log(π)
sample rInternal:add (noiseModule) -- Membuat noise
sampler: tambahkan (samplerInternal)
sampler: tambahkan (nn.CDadTable())
self.temperature = nn.MulConstant(1 / self.tau, true) -- Suhu untuk softmax
sampler: tambahkan (self.temperature)
sampler:add (nn.View(-1, self.k)) -- Ubah ukuran agar bekerja lebih dari k
sampler: tambahkan (nn.SoftMax())
sampler:add (nn.View(-1, self.N * self.k)) -- Ubah ukuran kembali
```