def __init__(self, hidden_dim, latent_dim, max_length):
    super(SketchVAELatentEmbedding, self).__init__()
    self.mu_embedding = nn.Linear(hidden_dim, latent_dim)
    self.sigma_embedding = nn.Linear(hidden_dim, latent_dim)
    self.gaussian_generator = MultivariateNormal(torch.zeros(latent_dim), torch.eye(latent_dim))
def __init__(self, hidden_dim, latent_dim, max_length):
    super(SketchVAELatentEmbedding, self).__init__()
    self.mu_embedding = nn.Linear(hidden_dim, latent_dim)
    self.sigma_embedding = nn.Linear(hidden_dim, latent_dim)
    self.gaussian_generator = MultivariateNormal(torch.zeros(latent_dim), torch.eye(latent_dim))