tensor add storage method (#10311)

对齐 torch 给tensor添加 storage 方法 --------- Co-authored-by: oneflow-ci-bot <[email protected]>
Oneflow-Inc · Aug 9, 2023 · f950a9a · f950a9a
1 parent 1ada508
commit f950a9a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/python/oneflow/framework/tensor.py b/python/oneflow/framework/tensor.py
@@ -533,6 +533,10 @@ def _conj_physical(self):
     return flow._C.conj_physical(self)
 
 
+def _storage(self):
+    return self
+
+
 @property
 def _layout(self):
     return flow.strided
@@ -610,6 +614,7 @@ def RegisterMethods():
     Tensor.conj = _conj
     Tensor.conj_physical = _conj_physical
     Tensor.layout = _layout
+    Tensor.storage = _storage
 
 
 def register_tensor_op(op_name):