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.