Python function

default_device

`default_device()`

max.experimental.tensor.default_device(device)

source

Context manager for setting the default device for tensor creation.

Sets the default device used for tensor creation within the context. All tensors created inside the context block without an explicit device parameter will use this device.

from max.experimental import tensor
from max.driver import CPU

with tensor.default_device(CPU()):
    x = tensor.Tensor.ones((2, 3))  # Created on CPU
    y = tensor.Tensor.zeros((2, 3))  # Also on CPU

Parameters:: device (Device | DeviceRef) – The device to use as the default for tensor creation within the context.
Returns:: A context manager that sets the default device.

default_device()​

`default_device()`