pytorch創建tensor的四種方法


import numpy as np
import torch

data = np.array([1,2,3])
type(data)

d1 = torch.Tensor(data)  #d1和data不共享,互相獨立,數據類型會默認為float64
print(d1)

d2 = torch.tensor(data)  #d2和data不共享,互相獨立,數據類型和data一致。常用
print(d2)

d3 = torch.as_tensor(data)  #data和d3內存共享,修改任何一個,另一個都會同步變化,數據類型和data一致。常用
print(d3) 

d4
= torch.from_numpy(data)  #data和d3內存共享,修改任何一個,另一個都會同步變化,數據類型和data一致,只能轉換numpy類型
print(d4)

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM