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