| 255 | |
| 256 | |
| 257 | class SingleVarianceNetwork(nn.Module): |
| 258 | def __init__(self, init_val): |
| 259 | super(SingleVarianceNetwork, self).__init__() |
| 260 | self.register_parameter('variance', nn.Parameter(torch.tensor(init_val))) |
| 261 | |
| 262 | def forward(self, x): |
| 263 | return torch.ones([len(x), 1]) * torch.exp(self.variance * 10.0) |