原文:torch.cuda.FloatTensor

Pytorch中的tensor又包括CPU上的數據類型和GPU上的數據類型,一般GPU上的Tensor是CPU上的Tensor加cuda 函數得到。 一般系統默認是torch.FloatTensor類型。例如data torch.Tensor , 是一個 的張量,類型為FloatTensor data.cuda 就轉換為GPU的張量類型,torch.cuda.FloatTensor類型。 ...

2020-02-24 15:54 0 5381 推薦指數:

查看詳情

torch.cuda.FloatTensor

Pytorch中的tensor又包括CPU上的數據類型和GPU上的數據類型,一般GPU上的Tensor是CPU上的Tensor加cuda()函數得到。 一般系統默認是torch.FloatTensor類型。例如data = torch.Tensor(2,3)是一個2*3的張量,類型 ...

Tue Dec 22 16:56:00 CST 2020 0 1141
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM