torch.Tensor.type_as

Tensor.type_as(tensor) → Tensor

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

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

Параметры

tensor (Тензор) – тензор, имеющий желаемый тип