GetDevice

Returns the device index of a tensor.

Abstract Signature:

GetDevice(input)

PyTorchJAX (Core)Apple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.get_device

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro '{input}.device()'

Apple MLX

API: —

Strategy: Macro 'mlx.core.default_device()'

Flax NNX

API: —

Strategy: Macro '{input}.device()'

PaxML / Praxis

API: —

Strategy: Macro '{input}.device()'