torch.Tensor.type

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

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

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

Параметры

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