Method get_device

dali/python/nvidia/dali/_tensor_formatting.py:62–64

Returns device string like 'CPU' or 'GPU'.

(self, obj: Any)

        ...

    def get_device(self, obj: Any) -> str:
        """Returns device string like 'CPU' or 'GPU'."""
        ...

    def get_dtype(self, obj: Any) -> Any:
        """Returns the data type."""