Fix device compatibility issue in VRComplex

aidos-lab / pytorch-topological

A topological machine learning framework based on PyTorch

https://pytorch-topological.readthedocs.io/

BSD 3-Clause "New" or "Revised" License

147 stars 21 forks source link

Closed crisbodnar closed 2 years ago

crisbodnar commented 2 years ago

Instantiate new tensors on the same device as the input tensor.

Pseudomanifold commented 2 years ago

Thanks! Will check other code for similar issues as well.