网站链接: element-ui dtcms
当前位置: 首页 > 技术博文  > 技术博文

pytorch tensor与numpy转换

2021/4/5 5:48:04 人评论

tensor to numpy a torch.ones(5) print(a) 输出 tensor([1., 1., 1., 1., 1.]) 进行转换 b a.numpy() print(b) 输出 [1. 1. 1. 1. 1.] 注意,转换后的tensor与numpy指向同一地址,所以,对一方的值改变另一方也随之改变 a.add_(1) …

tensor to numpy

a = torch.ones(5)
print(a)

输出

tensor([1., 1., 1., 1., 1.])

进行转换

b = a.numpy()
print(b)

输出

[1. 1. 1. 1. 1.]

注意,转换后的tensor与numpy指向同一地址,所以,对一方的值改变另一方也随之改变

a.add_(1)
print(a)
print(b)

numpy to tensor

import numpy as np
a = np.ones(5)
b = torch.from_numpy(a)
np.add(a, 1, out=a)
print(a)
print(b)

输出

[2. 2. 2. 2. 2.]
tensor([2., 2., 2., 2., 2.], dtype=torch.float64)

除chartensor外所有tensor都可以转换为numpy

 

 

承接Matlab、Python和C++的编程,机器学习、计算机视觉的理论实现及辅导,本科和硕士的均可,咸鱼交易,专业回答请走知乎,详谈请联系QQ号757160542,非诚勿扰。

 

相关资讯

    暂无相关的数据...

共有条评论 网友评论

验证码: 看不清楚?