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