pytorch學習筆記(Tensor,Variable,nn.Module,optim)

Tensor(張量) torch.Tensor 默認的是 torch.FloatTensor 數據類型,也可以定義我們自己想要的數據類型 當然也可以創建一個全是0的空的Tensor或者取一個正態分佈作爲隨機初始值 可以像numpy一樣通過索引的方式取得其中的元素,同時改變它的值 除此之外可以在Tensor和numpy.ndarray之間相互轉換: 判斷是否支持GPU,如果想把tensor放到GPU
相關文章
相關標籤/搜索