Skip to content

current_device

torch.tvarant.current_device() → int

Returns the current device index.

Return the current device index.

Returns:

Name Type Description
int int

Active device ordinal (0 today).
Source code in torch_tvarant/tvarant/__init__.py 
76
77
78
79
80
81
82
def current_device() -> int:
    """Return the current device index.

    Returns:
        int: Active device ordinal (``0`` today).
    """
    return int(_C.current_device())