torch.Tensor.cuda

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

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

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

Параметры:

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