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)
原文地址:https://www.cnblogs.com/Xycdada/p/13786581.html