功能:完成数组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'>
参考链接:
https://blog.csdn.net/qq_33345917/article/details/86552152