torch.Tensor.type_as

Tensor.type_as(tensor) → Tensor

将此张量转换为给定张量的类型。

如果张量已经是正确的类型，则此操作为无操作。这等效于 self.type(tensor.type())

参数

tensor (Tensor) – 具有所需类型的张量