def METHOD_NAME(self) -> torch.METHOD_NAME:
    METHOD_NAME = self._device
    # make this more explicit to always include the index
    if METHOD_NAME.type == "cuda" and METHOD_NAME.index is None:
        return torch.METHOD_NAME(f"cuda:{torch.cuda.current_device()}")
    return METHOD_NAME