torchoutil.core.make module

as_device(
device: Literal[None],
) → None[source]
as_device(
device: str | int | device = CUDA_IF_AVAILABLE,
) → device

Create concrete device object from device-like object.

as_dtype(
dtype: Literal[None] = None,
) → None[source]
as_dtype(
dtype: str | DTypeEnum | dtype,
) → dtype

Create concrete dtype object from dtype-like object.

as_generator(
generator: Literal[None] = None,
) → None[source]
as_generator(
generator: int | Generator | Literal['default'],
) → Generator

Create concrete generator object from generator-like object.

get_default_device() → device[source]

Returns default device used when creating a tensor.

get_default_dtype() → dtype[source]
get_default_generator() → Generator[source]
set_default_dtype(
dtype: dtype | None | Literal['default'] | str | DTypeEnum,
) → None[source]
set_default_generator(
generator: Generator | None | Literal['default'] | int,
) → None[source]