01-pytorch

Posted liu247

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了01-pytorch相关的知识，希望对你有一定的参考价值。

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的主要内容，如果未能解决你的问题，请参考以下文章