Closed Lkruitwagen closed 2 years ago
Addresses #109
n.b. I don't think Object classes inherit a self.device attribute? So I've set the zero tensor to acquire device from x.
Object
self.device
x
Closes #109.
Addresses #109
n.b. I don't think
Object
classes inherit aself.device
attribute? So I've set the zero tensor to acquire device fromx
.