Method __init__

diffpack/layer.py:68–70

(self, embedding_size=256, scale=1.0)

    /1618ddea340f3e4a2ed7852a0694a809775cf8d0/models/layerspp.py#L32 """

    def __init__(self, embedding_size=256, scale=1.0):
        super().__init__()
        self.W = nn.Parameter(torch.randn(embedding_size // 2) * scale, requires_grad=False)

    def forward(self, x):
        x *= self.W[None, :] * 2 * np.pi