(self, z: Tensor)
| 104 | return code_idx, None |
| 105 | |
| 106 | def decode(self, z: Tensor): |
| 107 | |
| 108 | x_d = self.quantizer.dequantize(z) |
| 109 | x_d = x_d.view(1, -1, self.code_dim).permute(0, 2, 1).contiguous() |
| 110 | |
| 111 | # decoder |
| 112 | x_decoder = self.decoder(x_d) |
| 113 | x_out = self.postprocess(x_decoder) |
| 114 | return x_out |
| 115 | |
| 116 | |
| 117 | class Encoder(nn.Module): |
no test coverage detected