def METHOD_NAME(self, z: torch.Tensor, use_sigmoid: bool = True) -> torch.Tensor:
    x = F.relu(self.decodeL(z))
    x = x.view(x.shape[0], self.channels[-1], *self.final_size)
    x = self.decode(x)
    if use_sigmoid:
        x = torch.sigmoid(x)
    return x