torch.Tensor.type

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

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

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

参数:

• dtype（dtype 或字符串）- 所需类型

• non_blocking (bool) – 如果 True ，并且源在固定内存中，目标在 GPU 上或反之，则异步复制相对于主机执行。否则，该参数无效果。

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