diff options
Diffstat (limited to 'scnn/scnn.py')
-rw-r--r-- | scnn/scnn.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/scnn/scnn.py b/scnn/scnn.py index a844a5a..2eae3f3 100644 --- a/scnn/scnn.py +++ b/scnn/scnn.py @@ -35,7 +35,7 @@ class SimplicialConvolution(nn.Module): assert(M == self.M) assert(C_in == self.C_in) - X = chebyshev.assemble(self.K, self.L, x) + X = scnn.chebyshev.assemble(self.K, self.L, x) y = torch.einsum("bimk,oik->bom", (X, self.theta)) assert(y.shape == (B, self.C_out, M)) |