torch.Tensor.type

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

如果未提供 dtype，則返回型別，否則將此物件轉換為指定型別。

如果已經是正確的型別，則不執行復制，並返回原始物件。

引數

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

• non_blocking (bool) – 如果設定為 True，並且源在固定的記憶體中而目標在 GPU 上（反之亦然），則複製相對於主機非同步執行。否則，該引數無效。

• **kwargs – 為了相容性，可能會包含 async 鍵來代替 non_blocking 引數。 async 引數已棄用。