torch.Tensor.type

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

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

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

引數

• dtype (dtype or string) – 期望的型別

• non_blocking (bool) – 如果為 True，且源位於鎖頁記憶體中，目標位於 GPU 上或反之，則複製操作相對於主機是非同步執行的。否則，此引數無效。

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