pytorch框架之张量TOC
张量与numpy数组之间的转换
torch的张量可以作为numpy数组的替代品,他们之间的相互转换十分方便,但是因为两者共享底层的内层地址,所以修改一个会导致另一个的变化。
Tenor转化为numpy数组:
import torch
import numpy
a = torch.ones(5)
b = a.numpy()
print(a,b)
#使用from_numpy将numpy数组转化为Tensor
b = torch.from_numpy(a)