torch.Tensor.type

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

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

如果此对象已是正确类型，则不执行复制并返回原始对象。

参数

• dtype (dtype or string) – 期望的类型

• non_blocking (bool) – 如果为 True，且源位于锁页内存中，目标位于 GPU 上或反之，则复制操作相对于主机是异步执行的。否则，此参数无效。

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