torch.Tensor.type_as

Tensor.type_as(tensor) → Tensor

Возвращает этот тензор, преобразованный к типу заданного тензора.

Если тензор уже имеет нужный тип, операция ничего не делает. Это эквивалентно self.type(tensor.type())

Параметры:

tensor (Tensor) – тензор, тип которого требуется получить