bump to 0.1.40: fix formula

leibniz/nn/layer/hyperbolic.py

@@ -57,7 +57,7 @@ def __init__(self, dim, step, relu, conv, reduction=16):
         self.step = step
 
         self.input = BasicBlock(dim, 2 * dim, step, relu, conv, reduction=reduction)
-        self.output = BasicBlock(4 * dim, dim, step, relu, conv, reduction=reduction)
+        self.output = BasicBlock(5 * dim, dim, step, relu, conv, reduction=reduction)
 
     def forward(self, x):
         input = self.input(x) * self.step
@@ -68,7 +68,7 @@ def forward(self, x):
         y2 = x * u - v
         y3 = - x * v - u
         y4 = - x * u + v
-        ys = th.cat((y1, y2, y3, y4), dim=1)
+        ys = th.cat((y1, y2, y3, y4, x), dim=1)
 
         return x + self.output(ys)
 
@@ -83,7 +83,7 @@ def __init__(self, dim, step, relu, conv, reduction=16):
         self.step = step
 
         self.input = Bottleneck(dim, 2 * dim, step, relu, conv, reduction=reduction)
-        self.output = Bottleneck(4 * dim, dim, step, relu, conv, reduction=reduction)
+        self.output = Bottleneck(5 * dim, dim, step, relu, conv, reduction=reduction)
 
     def forward(self, x):
         input = self.input(x) * self.step
@@ -94,6 +94,6 @@ def forward(self, x):
         y2 = x * u - v
         y3 = - x * v - u
         y4 = - x * u + v
-        ys = th.cat((y1, y2, y3, y4), dim=1)
+        ys = th.cat((y1, y2, y3, y4, x), dim=1)
 
         return x + self.output(ys)

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="leibniz",
-    version="0.1.39",
+    version="0.1.40",
     author="Mingli Yuan",
     author_email="[email protected]",
     description="Leibniz is a package providing facilities to express learnable differential equations based on PyTorch",