chainerx.using_device¶
- chainerx.using_device(device)¶
Creates a context manager to temporarily set the default device.
- Parameters
device (Device or str) – Device object or device name to set as the default device during the context. See
chainerx.Device.name
for the specification of device names.