Torchebm > Core > Device_mixin¶
Contents¶
Classes¶
DeviceMixin- A mixin for consistent device and dtype management across all modules.
Functions¶
normalize_device()- Normalizes the device identifier for consistent usage.
API Reference¶
torchebm.core.device_mixin ¶
This module handles the device management or TorchEBM modules