torch.Tensor.type

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

如果没有提供dtype，则返回类型；否则，将此对象转换为指定的类型。

如果已经是正确类型，就不会进行复制，直接返回原对象。

参数

• dtype (dtype 或 字符串) – 指定所需的类型

• non_blocking (bool) – 如果为 True，且源在 pinned 内存中而目标在 GPU 上或反之，则复制操作将异步执行。否则，此参数无效。

• **kwargs – 为了兼容性，可能包含async关键字来代替non_blocking参数。请注意，async已经弃用。