sampler:add(samplerInternal) sampler:add(nn.CAddTable()) self.temperature = nn.MulConstant(1 / self.tau, true) -- Temperature τ for softmax sampler:add(self.temperature) sampler:add(nn.View(-1, self.k)) -- Resize to work over k sampler:add(nn.SoftMax()) sampler:add(nn.View(-1...