首页 > 其他 > 详细

01-pytorch

时间:2019-07-07 12:20:34      阅读:93      评论:0      收藏:0      [点我收藏+]
import torch
import numpy as np

np和torch互相转换

np_data = np.arange(6).reshape((2,3))
#np ---> torch
torch_data = torch.from_numpy(np_data)
#torch --> np
torch_data_2= torch_data.numpy()

[[0 1 2]
[3 4 5]]
tensor([[0, 1, 2],
[3, 4, 5]])

torch的运算

求绝对值
torch.abs(tensor)
矩阵向乘
torch.mm(tensor,tensor)

```python
data = [-1,-2,1,2]
tensor = torch.FloatTensor(data)
torch.abs(tensor)

01-pytorch

原文:https://www.cnblogs.com/liu247/p/11145504.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!