ran isort

isort --recursive --atomic --apply pts
2026-07-14 11:17:47 +08:00 · 2019-10-30 09:38:19 +01:00
parent 52ed9d8937
commit 4ad01ea2e3
25 changed files with 79 additions and 97 deletions
@@ -0,0 +1,11 @@
+import torch
+import torch.nn as nn
+
+
+class LambdaLayer(nn.Module):
+    def __init__(self, function):
+        super().__init__()
+        self._func = function
+
+    def forward(self, x, *args):
+        return self._func(x, *args)