summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGard Spreemann <gspreemann@gmail.com>2018-10-25 20:14:09 +0200
committerGard Spreemann <gspreemann@gmail.com>2018-10-25 20:14:09 +0200
commit4a3b1f8951388086b2c28d9a17426ce4085480cd (patch)
treed5b061c60915170693293e93e62039c6ec0af4de
parenta1399a3e0556abb6ea9703e7503bf954e8ab0514 (diff)
Correct namespace.
-rw-r--r--scnn/scnn.py2
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))