quaterion_models.utils.tensors module
- move_to_device(obj, device: device)[source]
Given a structure (possibly) containing Tensors on the CPU, move all the Tensors to the specified GPU (or do nothing, if they should be on the CPU).
Given a structure (possibly) containing Tensors on the CPU, move all the Tensors to the specified GPU (or do nothing, if they should be on the CPU).
Explore practical problem solving with Similarity Learning
Learn Similarity LearningFind people dealing with similar problems and get answers to your questions
Join CommunityTo analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. Read our Privacy Policy.
