numpy和tensor互转

  • Post author:
  • Post category:其他


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 版权协议,转载请附上原文出处链接和本声明。