Pytorch: 검벨 노이즈

안녕,

여기 에서와 같이 Gumbel 노이즈를 추가하려고 시도했지만 성공하지 못했습니다.
여러 모듈이 여전히 누락된 것 같습니까(예: nn.Uniform()과 같은 기본 변수 함수), 아니면 제가 잘못 알고 있습니까? 예를 들어 다음 행과 같이 pytorch에서 구현하는 방법은 다음과 같습니다.

``` -- 노이즈 ε 샘플 모듈 생성
로컬 노이즈 모듈 = nn.Sequential()
no iseModule:add (nn.Uniform(0, 1)) -- U(0, 1)의 샘플
-- 균일한 샘플을 Gumbel 샘플로 변환
no iseModule:add (nn.AddConstant(1e-9, true)) -- 수치적 안정성 향상
iseModule: 추가 (nn.Log()) 없음
iseModule: 추가 (nn.MulConstant(-1, true)) 없음
no iseModule:add (nn.AddConstant(1e-9, true)) -- 수치적 안정성 향상
iseModule: 추가 (nn.Log()) 없음
iseModule: 추가 (nn.MulConstant(-1, true)) 없음

-- 샘플러 생성 q(z) = G(z) = softmax((log(π) + ε)/τ) (재매개변수화 트릭)
로컬 샘플러 = nn.Sequential()
로컬 샘플러 내부 = nn.ConcatTable()
sample rInternal:add (nn.Identity()) -- 정규화되지 않은 로그 확률 log(π)
sample rInternal:add (noiseModule) -- 노이즈 ε 생성
샘플러:추가 (샘플러 내부)
샘플러:추가 (nn.CAddTable())
self.temperature = nn.MulConstant(1 / self.tau, true) -- softmax의 온도 τ
샘플러:추가 (자체 온도)
sampler:add (nn.View(-1, self.k)) -- k에서 작동하도록 크기 조정
샘플러:추가 (nn.SoftMax())
sampler:add (nn.View(-1, self.N * self.k)) -- 다시 크기 조정
```