torch.Tensor.type

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

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

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

Parameters

• dtype (dtype 或 字符串) – 所需的数据类型

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

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