1 torch.from_nump()
功能:完成数组numpy到tensor的转换。
栗子:
import torch
import numpy as np
a = np.ones(5)
b = torch.from_numpy(a)
print(type(a))
print(type(b))
输出结果:
<class 'numpy.ndarray'>
<class 'torch.Tensor'>
注意:
这个函数所产生的的Tensor和NumPy中的数组共享相同的内存(所以他们之间的转换很快),改变其中一个时另一个也会改变!!!
版权声明:本文为weixin_40522801原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。