Support tensor storage for compat interface

tat/compat.py

@@ -275,11 +275,14 @@ def _(self: T, position: dict[str, tuple[typing.Any, int]]) -> tuple[int, ...]:
 # Function renames
 
 
-def _compat_function(focus_type: type) -> typing.Callable[[typing.Callable], typing.Callable]:
+def _compat_function(focus_type: type, name: str | None = None) -> typing.Callable[[typing.Callable], typing.Callable]:
 
     def _result(function: typing.Callable) -> typing.Callable:
-        name = function.__name__
-        setattr(focus_type, name, function)
+        if name is None:
+            attr_name = function.__name__
+        else:
+            attr_name = name
+        setattr(focus_type, attr_name, function)
         return function
 
     return _result
@@ -303,6 +306,14 @@ def zero(self: T) -> T:
     return self.zero_()
 
 
+@_compat_function(T, name="storage")  # type: ignore[misc]
+@property
+def storage(self: T) -> typing.Any:
+    "Get the storage of the tensor"
+    assert self.data.is_contiguous()
+    return self.data.numpy().reshape([-1])
+
+
 # Exponential arguments
 
 origin_exponential = T.exponential