torch.Tensor.xpu

Tensor.xpu(device=None, non_blocking=False, memory_format=torch.preserve_format) → Tensor

Возвращает копию этого объекта в памяти XPU.

Если этот объект уже находится в памяти XPU и на нужном устройстве, копирование не выполняется, и возвращается исходный объект.

Параметры:

• device (torch.device, optional) – Целевое устройство XPU. По умолчанию используется текущее устройство XPU.
• non_blocking (bool, optional) – Если True и источник находится в закреплённой памяти, копирование будет выполняться асинхронно относительно хоста. В противном случае этот аргумент не влияет на выполнение. Значение по умолчанию: False.
• memory_format (torch.memory_format, optional) – желаемый формат памяти возвращаемого Tensor. Значение по умолчанию: torch.preserve_format.