torchwrench.core.make module¶

torchwrench.core.make.as_device(device: None) → None[source]¶

torchwrench.core.make.as_device(device: str | int | device = CUDA_IF_AVAILABLE) → device

Create concrete device object from device-like object.

torchwrench.core.make.as_dtype(dtype: None = None) → None[source]¶

torchwrench.core.make.as_dtype(dtype: str | DTypeEnum | dtype) → dtype

Create concrete dtype object from dtype-like object.

torchwrench.core.make.as_generator(generator: None = None) → None[source]¶

torchwrench.core.make.as_generator(generator: int | Generator | 'default') → Generator

Create concrete generator object from generator-like object.

torchwrench.core.make.get_default_device() → device[source]¶

Returns default device used when creating a tensor.

torchwrench.core.make.get_default_dtype() → dtype[source]¶

torchwrench.core.make.get_default_generator() → Generator[source]¶

torchwrench.core.make.set_default_dtype(dtype: dtype | None | 'default' | str | DTypeEnum) → None[source]¶

torchwrench.core.make.set_default_generator(generator: Generator | None | 'default' | int) → None[source]¶