tensor add storage method

Oneflow-Inc · Aug 8, 2023 · 3e63f9d · 3e63f9d
1 parent b664acc
commit 3e63f9d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/oneflow/framework/tensor.py b/python/oneflow/framework/tensor.py
@@ -532,6 +532,8 @@ def _conj(self):
 def _conj_physical(self):
  return flow._C.conj_physical(self)
 
+def _storage(self):
+ return self
 
 @property
 def _layout(self):
@@ -610,6 +612,7 @@ def RegisterMethods():
  Tensor.conj = _conj
  Tensor.conj_physical = _conj_physical
  Tensor.layout = _layout
+ Tensor.storage = _storage
 
 
 def register_tensor_op(op_name):