Context & Defaults

defaults

def defaults(dtype: 'DType | None' = None, device: 'Device | None' = None) -> 'tuple[DType, Device]':

Get default dtype and device for tensor creation.

---

default_device

def default_device(device: 'Device | graph.DeviceRef') -> 'Generator[None, None, None]':

Context manager setting default device.

---

default_dtype

def default_dtype(dtype: 'DType') -> 'Generator[None, None, None]':

Context manager setting default dtype.

---

defaults_like

def defaults_like(like: 'Tensor | TensorType') -> 'Generator[None, None, None]':

Context manager setting default dtype and device from template.

---