small additions (#130)

* add torch.fft dummy * add Tensor.cuda * Tensor.flatten allow start and end dim
rwth-i6 · Dec 7, 2022 · 5c17327 · 5c17327
1 parent 8fa7e93
commit 5c17327
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/pytorch_to_returnn/torch/fft.py b/pytorch_to_returnn/torch/fft.py
diff --git a/pytorch_to_returnn/torch/tensor.py b/pytorch_to_returnn/torch/tensor.py
@@ -98,15 +98,18 @@ def clone(self):
   def cpu(self):
     return self  # ignore
 
+  def cuda(self):
+    return self  # ignore
+
   @property
   def device(self):
     class DeviceDummy:
       type = None
     return DeviceDummy()
 
-  def flatten(self):
+  def flatten(self, start_dim=0, end_dim=-1):
     from .nn.functional import flatten
-    return flatten(self)
+    return flatten(self, start_dim, end_dim)
 
   def view(self, *shape):
     from .nn.functional import reshape