fix old codes

2022-07-11 10:58:15 +02:00
parent 088649e042
commit f526a26050
19 changed files with 385 additions and 457 deletions
--- a/tests/test_fnn.py
+++ b/tests/test_fnn.py
@@ -12,7 +12,7 @@ class myFeature(torch.nn.Module):
        super(myFeature, self).__init__()

    def forward(self, x):
-        return torch.sin(torch.pi * x.extract('a'))
+        return LabelTensor(torch.sin(torch.pi * x.extract('a')), 'sin(a)')


 data = torch.rand((20, 3))