torch.Tensor.type

Tensor.type(dtype=None, non_blocking=False, **kwargs) → str or Tensor

Возвращает тип, если dtype не указан, в противном случае преобразует этот объект к указанному типу.

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

Параметры:

• dtype (тип или строка) – Желаемый тип
• non_blocking (bool) – Если True, и источник находится в закреплённой памяти, а назначение — на GPU или наоборот, копирование выполняется асинхронно относительно хоста. В противном случае аргумент не имеет эффекта.
• **kwargs – Для совместимости может содержать ключ async вместо аргумента non_blocking. Аргумент async устарел.