torch.Tensor.type

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

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

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

Параметры:

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