Merge pull request #3899 from ggosline/pytorch_2.0_compiler

Pytorch 2.0 compiler
fastai · Mar 28, 2023 · 125d2ef · 125d2ef
2 parents 930d3e9 + 909b673
commit 125d2ef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/fastai/layers.py b/fastai/layers.py
@@ -68,7 +68,7 @@ def __repr__(self): return f'{self.__class__.__name__}({self.repr})'
 @module(full=False)
 def Flatten(self, x):
     "Flatten `x` to a single dimension, e.g. at end of a model. `full` for rank-1 tensor"
-    return TensorBase(x.view(-1) if self.full else x.view(x.size(0), -1))
+    return x.view(-1) if self.full else x.view(x.size(0), -1)  # Removed cast to Tensorbase
 
 # %% ../nbs/01_layers.ipynb 15
 @module(tensor_cls=TensorBase)

diff --git a/nbs/01_layers.ipynb b/nbs/01_layers.ipynb
@@ -190,7 +190,7 @@
     "@module(full=False)\n",
     "def Flatten(self, x):\n",
     "    \"Flatten `x` to a single dimension, e.g. at end of a model. `full` for rank-1 tensor\"\n",
-    "    return TensorBase(x.view(-1) if self.full else x.view(x.size(0), -1))"
+    "    return x.view(-1) if self.full else x.view(x.size(0), -1)  # Removed cast to Tensorbase"
    ]
   },
   {