Merge pull request #415 from Jittor/from_torch

add from_torch
2022-12-02 22:00:39 +08:00 · 2022-12-02 22:00:39 +08:00 · fe6970efc8
parent be8faf4dfc 725615872a
commit fe6970efc8
1 changed files with 6 additions and 0 deletions
--- a/python/jittor/misc.py
+++ b/python/jittor/misc.py
@ -1985,3 +1985,9 @@ def cpu(x): return x.clone()
 jt.Var.cpu = cpu
 def to(x, *args, **kargs): return x.clone()
 jt.Var.to = to
+
+def from_torch(x):
+    '''
+    Convert torch Tensor to Jittor Var
+    '''
+    return jt.Var(x.cpu().numpy())