(self, x, sample_idx=None)
| 307 | return x |
| 308 | |
| 309 | def decode(self, x, sample_idx=None): |
| 310 | x = (x * (self.std + self.eps)) + self.mean |
| 311 | return x |
| 312 | |
| 313 | def cuda(self): |
| 314 | self.mean = self.mean.cuda() |
nothing calls this directly
no outgoing calls
no test coverage detected