We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 15c166a commit 44c2ac2Copy full SHA for 44c2ac2
torchvision/tv_tensors/__init__.py
@@ -8,6 +8,10 @@
8
from ._video import Video
9
10
11
+# TODO: Fix this. We skip this method as it leads to
12
+# RecursionError: maximum recursion depth exceeded while calling a Python object
13
+# Keeping it here, leads to graph breaks between multiple functional ops instead of having a single graph
14
+@torch.compiler.disable
15
def wrap(wrappee, *, like, **kwargs):
16
"""[BETA] Convert a :class:`torch.Tensor` (``wrappee``) into the same :class:`~torchvision.tv_tensors.TVTensor` subclass as ``like``.
17
0 commit comments