torch.Tensor.type

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

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

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

参数

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

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

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