Commit bc06cd4a authored by Lukas Miklautz's avatar Lukas Miklautz
Browse files

Update acedec.py by removing redundant code

parent da64c22e
......@@ -41,30 +41,6 @@ class AcedecFramework(torch.nn.Module):
z_rot = self.rotate(z)
return z_rot
def encode_clustered_space(self, x:torch.Tensor)->torch.Tensor:
"""
Args:
x: input data point, can also be a mini-batch of points
Returns:
embedded_cs: the embedded times beta_cs data point with dimensionality embedding_size
"""
z_rot = self.encode(x)
z_cs = z_rot * self.subspace_betas()[0]
return z_cs
def encode_noise_space(self, x:torch.Tensor)->torch.Tensor:
"""
Args:
x: input data point, can also be a mini-batch of points
Returns:
embedded_ns: the embedded data point times beta_ns with dimensionality embedding_size
"""
z_rot = self.encode(x)
z_ns = z_rot * self.subspace_betas()[1]
return z_ns
def decode(self, embedded:torch.Tensor)->torch.Tensor:
"""
Args:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment