* Adding title kwarg for plotter class
* Create tutorial for Fourier Feature Embedding * Update doc for tutorials
This commit is contained in:
committed by
Nicola Demo
parent
4b64998f45
commit
e514969a43
@@ -235,7 +235,7 @@ class FourierFeatureEmbedding(torch.nn.Module):
|
||||
size = (input_dimension,
|
||||
output_dimension // 2),
|
||||
requires_grad = False
|
||||
)
|
||||
) * self.sigma
|
||||
|
||||
def forward(self, x):
|
||||
"""
|
||||
@@ -248,7 +248,7 @@ class FourierFeatureEmbedding(torch.nn.Module):
|
||||
# compute random matrix multiplication
|
||||
out = torch.mm(x, self._matrix)
|
||||
# return embedding
|
||||
return torch.cat([torch.cos(out), torch.sin(out)], dim=-1)
|
||||
return torch.cat([torch.cos(2*torch.pi*out), torch.sin(2*torch.pi*out)], dim=-1)
|
||||
|
||||
|
||||
@property
|
||||
|
||||
Reference in New Issue
Block a user