Pytorch: Gumbel-Geräusch

Hallo,

Ich habe versucht, Gumbel-Rauschen wie hier hinzuzufügen, aber ohne Erfolg.
Sieht so aus, als würden noch einige Module fehlen (z. B. grundlegende Variablenfunktionen wie nn.Uniform()), oder irre ich mich? Wie würden Sie in Pytorch zB die folgenden Zeilen implementieren:

``` -- Rauschen erzeugen ε Beispielmodul
lokales Rauschmodul = nn.Sequential()
no iseModule:add (nn.Uniform(0, 1)) – Sample von U(0, 1)
-- Gleichförmige Probe in Gumbel-Probe umwandeln
no iseModule:add (nn.AddConstant(1e-9, true)) – Verbessert die numerische Stabilität
nein iseModule:add (nn.Log())
nein iseModule:add (nn.MulConstant(-1, true))
no iseModule:add (nn.AddConstant(1e-9, true)) – Verbessert die numerische Stabilität
nein iseModule:add (nn.Log())
nein iseModule:add (nn.MulConstant(-1, true))

-- Sampler erstellen q(z) = G(z) = softmax((log(π) + ε)/τ) (Reparametrisierungstrick)
lokaler Sampler = nn.Sequential()
lokaler SamplerInternal = nn.ConcatTable()
sample rInternal:add (nn.Identity()) -- Unnormalisierte Log-Wahrscheinlichkeiten log(π)
sample rInternal:add (noiseModule) – Erzeuge Rauschen ε
sampler:add (samplerInternal)
sampler:add (nn.CAddTable())
self.temperature = nn.MulConstant(1 / self.tau, true) – Temperatur τ für softmax
sampler:add (selbst.temperatur)
sampler:add (nn.View(-1, self.k)) – Ändere die Größe, um über k zu arbeiten
sampler:add (nn.SoftMax())
sampler:add (nn.View(-1, self.N * self.k)) – Resize back
```