(self, input_tensor)
| 139 | return z, mu, log_sigma |
| 140 | |
| 141 | def encoder(self, input_tensor): |
| 142 | z_ = self.emb_encode(input_tensor) |
| 143 | z = self.aggregate(z_) |
| 144 | self.z, mu, log_sigma = self.z_encode(z) |
| 145 | return self.z, mu, log_sigma |
| 146 | |
| 147 | def forward(self, input_tensor): |
| 148 | self.z, _, _ = self.encoder(input_tensor) |
no test coverage detected