torch.Tensor.type

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

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

如果当前类型已是目标类型，则不会执行任何拷贝操作，并直接返回原对象。

参数:

• dtype (dtype 或 字符串) – 期望的类型

• non_blocking (bool) – 如果设置为 True，且源内存位于固定内存（pinned memory）而目标在 GPU 上（反之亦然），则拷贝将相对于主机异步执行。否则，该参数无效。

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