jittor
jittor copied to clipboard
How to integrate with torch data?
If a torch data is
a = torch.rand(3)
b = jt.float(a)
this can convert torch data to jittor, but how to convert jittor data to torch ?
Firstly, convert jittor var to numpy. Then you can use it in torch.
a = torch.rand(3)
b = jt.float(a)
c = b.data # numpy array
d = torch.from_numpy(c)