在pytorch中,把numpy.array数据转换到张量tensor数据的常用函数是torch.from_numpy(array)或者torch.Tensor(array),第一种函数更常用,然而在pytorch0.4中已经舍弃了这种函数,下面一个简单的编程实验说明这两种方法的区别,实验在pytorch0.4框架下进行
运行程序之后,结果是
可以看出修改数组a的元素值,张量b的元素值也改变了,但是张量c却不变。修改张量c的元素值,数组a和张量b的元素值都不变。这说明torch.from_numpy(array)是做数组的浅拷贝,torch.Tensor(array)是做数组的深拷贝
热门工具 换一换